Update inference.py

inference.py

@@ -60,8 +60,9 @@ class ShellDConfig:
     def from_json(cls, path: str) -> "ShellDConfig":
         with open(path) as f:
             d = json.load(f)
-        # Strip keys starting with "_"
-        d = {k: v for k, v in d.items() if not k.startswith("_")}
+        # Keep only fields that exist in the dataclass
+        valid_keys = {f.name for f in cls.__dataclass_fields__.values()}
+        d = {k: v for k, v in d.items() if k in valid_keys}
         return cls(**d)