allow?

inference.py

@@ -133,6 +133,7 @@ def make_inference(data_dict:list) -> str:
     test_ds = Dataset(
         data=data_dict,
         transform=transforms,
+        allow_missing_keys=True,
     )
     model.eval()
     with torch.no_grad():