diff --git a/omnigibson/systems/__init__.py b/omnigibson/systems/__init__.py index 7df38267d..fe8f42afc 100644 --- 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", ]