diff --git a/tests/engines/test_patche_inference_engine.py b/tests/engines/test_patche_inference_engine.py index 7caae59..ac4ba77 100644 --- a/tests/engines/test_patche_inference_engine.py +++ b/tests/engines/test_patche_inference_engine.py @@ -1,6 +1,6 @@ import torch import lightning as L -from minerva.utils.patches import WeightedAvgPatchInferencer, VotingPatchInferencer +from minerva.emgines.patch_inferencer_engine import WeightedAvgPatchInferencer, VotingPatchInferencer pyramid = [ [