Fixes

Lightning-Universe · Sep 3, 2021 · db70df5 · db70df5
1 parent 9ac1048
commit db70df5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/image/detection/test_model.py b/tests/image/detection/test_model.py
@@ -78,7 +78,7 @@ def test_init():
     dl = model.process_predict_dataset(ds, batch_size=batch_size)
     data = next(iter(dl))
 
-    out = model(data)
+    out = model.forward(data[DefaultDataKeys.INPUT])
 
     assert len(out) == batch_size
     assert all(isinstance(res, dict) for res in out)