def test_dl_latest_saved_model_no_path(self): with self.assertRaises(InvalidQueryError): client = Client( api_token="57c60ade109be36ef1a1c89f56247109fa448741") client.checkout_project( project_token="4b003477-3b31-4f74-8952-8a9dc879b0ec") client.checkout_network(network_name="ssd_inceptionV2_COCO") client.dl_latest_saved_model()
def test_dl_latest_saved_model_path(self): client = Client(api_token="57c60ade109be36ef1a1c89f56247109fa448741") client.checkout_project( project_token="4b003477-3b31-4f74-8952-8a9dc879b0ec") client.checkout_network(network_name="ssd_inceptionV2_COCO") self.assertEqual(client.network_id, "ff3fb57d-6f95-46f6-a6dc-aeb0985dd6a4") client.dl_latest_saved_model(path_to_save="test_saved/ici") self.assertTrue(os.path.isdir("test_saved/ici")) self.assertTrue( os.path.isfile(os.path.join("test_saved/ici", "saved_model.pb"))) shutil.rmtree("test_saved/ici")