everything setup up with logger and runtime typechecking

MolarVerse · May 12, 2024 · 9e98161 · 9e98161
1 parent 1c761d4
commit 9e98161
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/PQAnalysis/traj/trajectory.py b/PQAnalysis/traj/trajectory.py
@@ -9,7 +9,7 @@
 
 from PQAnalysis.topology import Topology
 from PQAnalysis.types import Np2DNumberArray, Np1DNumberArray
-from PQAnalysis.exceptions import PQIndexError
+from PQAnalysis.exceptions import PQIndexError, PQTypeError
 from PQAnalysis.core import Cell
 from PQAnalysis.atomic_system import AtomicSystem
 from PQAnalysis.utils.custom_logging import setup_logger
@@ -309,7 +309,6 @@ def __contains__(self, item: AtomicSystem) -> bool:
         """
         return item in self.frames
 
-    @runtime_type_checking
     def __add__(self, other: "Trajectory") -> "Trajectory":
         """
         This method allows two trajectories to be added together.
@@ -320,6 +319,12 @@ def __add__(self, other: "Trajectory") -> "Trajectory":
             The other trajectory to add.
         """
 
+        if not isinstance(other, Trajectory):
+            self.logger.error(
+                "Only Trajectory objects can be added to a Trajectory.",
+                exception=PQTypeError,
+            )
+
         return Trajectory(self.frames + other.frames)
 
     def __eq__(self, other: Any) -> bool: