Update mpt references after relocation

Author: relh

packages/mettagrid/python/src/mettagrid/util/uri_resolvers/schemes.py

@@ -285,7 +285,7 @@ def policy_spec_from_uri(
     if parsed.canonical.endswith(".mpt"):
         checkpoint_path = str(parsed.local_path) if parsed.local_path else parsed.canonical
         return PolicySpec(
-            class_path="mettagrid.policy.mpt_policy.MptPolicy",
+            class_path="metta.rl.mpt_policy.MptPolicy",
             init_kwargs={
                 "checkpoint_uri": checkpoint_path,
                 "device": device,

recipes/experiment/cvc/mission_variant_curriculum.py

@@ -555,7 +555,7 @@ def _get_policy_action_space(policy_uri: str) -> Optional[int]:
         return None
 
     try:
-        from mettagrid.policy.mpt_artifact import load_mpt
+        from metta.rl.mpt_artifact import load_mpt
 
         artifact = load_mpt(policy_uri)
 

tests/rl/test_policy_artifact.py

@@ -13,7 +13,7 @@
 from metta.agent.policies.vit import ViTDefaultConfig
 from metta.agent.policy import Policy, PolicyArchitecture
 from mettagrid.base_config import Config
-from mettagrid.policy.mpt_artifact import MptArtifact, load_mpt, save_mpt
+from metta.rl.mpt_artifact import MptArtifact, load_mpt, save_mpt
 from mettagrid.policy.policy_env_interface import PolicyEnvInterface