Add inits

lerobot/common/policies/__init__.py

@@ -12,8 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from .act.configuration_act import ACTConfig as ACTConfig
-from .diffusion.configuration_diffusion import DiffusionConfig as DiffusionConfig
-from .pi0.configuration_pi0 import PI0Config as PI0Config
-from .tdmpc.configuration_tdmpc import TDMPCConfig as TDMPCConfig
-from .vqbet.configuration_vqbet import VQBeTConfig as VQBeTConfig
+from . import act, diffusion, pi0, tdmpc, vqbet
+from .factory import make_policy
+
+__all__ = ["act", "diffusion", "pi0", "tdmpc", "vqbet", "make_policy"]

lerobot/common/policies/act/__init__.py

@@ -0,0 +1,4 @@
+from .configuration_act import ACTConfig
+from .modeling_act import ACT
+
+__all__ = ["ACTConfig", "ACT"]

lerobot/common/policies/diffusion/__init__.py

@@ -0,0 +1,4 @@
+from .configuration_diffusion import DiffusionConfig
+from .modeling_diffusion import DiffusionPolicy
+
+__all__ = ["DiffusionConfig", "DiffusionPolicy"]

lerobot/common/policies/pi0/__init__.py

@@ -0,0 +1,4 @@
+from .configuration_pi0 import PI0Config
+from .modeling_pi0 import PI0Policy
+
+__all__ = ["PI0Config", "PI0Policy"]

lerobot/common/policies/tdmpc/__init__.py

@@ -0,0 +1,4 @@
+from .configuration_tdmpc import TDMPCConfig
+from .modeling_tdmpc import TDMPCPolicy
+
+__all__ = ["TDMPCConfig", "TDMPCPolicy"]

lerobot/common/policies/vqbet/__init__.py

@@ -0,0 +1,4 @@
+from .configuration_vqbet import VQBeTConfig
+from .modeling_vqbet import VQBeTPolicy
+
+__all__ = ["VQBeTConfig", "VQBeTPolicy"]