move hoomd import inside rigid body function (#80)

cmelab · Feb 16, 2024 · 6a68c97 · 6a68c97
1 parent 4a627bb
commit 6a68c97
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cmeutils/gsd_utils.py b/cmeutils/gsd_utils.py
@@ -3,7 +3,6 @@
 
 import freud
 import gsd.hoomd
-import hoomd
 import networkx as nx
 import numpy as np
 from boltons.setutils import IndexedSet
@@ -255,6 +254,8 @@ def update_rigid_snapshot(snapshot, mb_compound):
         of the complete system
 
     """
+    import hoomd
+
     rigid_ids = [p.rigid_id for p in mb_compound.particles()]
     rigid_bodies = set(rigid_ids)
     # Total number of rigid body particles