Add frames property to trajectory objects

klarh · Jul 24, 2019 · 6a315c8 · 6a315c8
1 parent 90cf06d
commit 6a315c8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/keras_gtar/trajectory.py b/keras_gtar/trajectory.py
@@ -31,8 +31,12 @@ def __exit__(self, typ, val, trace):
         self.close()
 
     def __len__(self):
+        return len(self.frames)
+
+    @property
+    def frames(self):
         (_, frames) = self.traj.framesWithRecordsNamed('weight')
-        return len(frames)
+        return frames
 
     def close(self):
         self.traj.close()