Add missing systems

StanfordVL · Dec 27, 2024 · 9fb980e · 9fb980e
1 parent 6c3f232
commit 9fb980e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/omnigibson/systems/__init__.py b/omnigibson/systems/__init__.py
@@ -1,3 +1,4 @@
+from omnigibson.systems.system_base import BaseSystem, PhysicalParticleSystem, VisualParticleSystem
 from omnigibson.systems.macro_particle_system import (
     MacroParticleSystem,
     MacroPhysicalParticleSystem,
@@ -12,6 +13,7 @@
 )
 
 __all__ = [
+    "BaseSystem",
     "Cloth",
     "FluidSystem",
     "GranularSystem",
@@ -20,4 +22,6 @@
     "MacroVisualParticleSystem",
     "MicroParticleSystem",
     "MicroPhysicalParticleSystem",
+    "PhysicalParticleSystem",
+    "VisualParticleSystem",
 ]