Search for output model in ./training rather than ./framework (#974)

KSGulin · web-flow · commit 43aafc7defe7 · 2022-08-05T13:32:13.000-04:00
diff --git a/tests/integrations/image_classification/test_image_classification.py b/tests/integrations/image_classification/test_image_classification.py
@@ -75,7 +75,7 @@ def capture_pre_run_state(self):
             self.expected_checkpoint_path = os.path.join(
                 train_args.save_dir,
                 train_args.model_tag,
-                "framework",
+                "training",
                 "model-one-shot.pth" if train_args.one_shot else "model.pth",
             )
 

Original file line number	Diff line number	Diff line change
`@@ -75,7 +75,7 @@ def capture_pre_run_state(self):`
`75`	`75`	`self.expected_checkpoint_path = os.path.join(`
`76`	`76`	`train_args.save_dir,`
`77`	`77`	`train_args.model_tag,`
`78`		`- "framework",`
	`78`	`+ "training",`
`79`	`79`	`"model-one-shot.pth" if train_args.one_shot else "model.pth",`
`80`	`80`	`)`
`81`	`81`