update tests

VipulMascarenhas · VipulMascarenhas · commit a185a3e7d047 · 2024-05-29T14:45:46.000-07:00
diff --git a/ads/aqua/extension/model_handler.py b/ads/aqua/extension/model_handler.py
@@ -100,8 +100,6 @@ class AquaModelLicenseHandler(AquaAPIhandler):
     @handle_exceptions
     def get(self, model_id):
         """Handle GET request."""
-
-        model_id = model_id.split("/")[0]
         return self.finish(AquaModelApp().load_license(model_id))
 
 
diff --git a/tests/unitary/with_extras/aqua/test_model_handler.py b/tests/unitary/with_extras/aqua/test_model_handler.py
@@ -75,7 +75,7 @@ def test_list(self, mock_list):
             )
 
     @patch("notebook.base.handlers.APIHandler.finish")
-    @patch("ads.aqua.model.model.AquaModelApp.register")
+    @patch("ads.aqua.model.AquaModelApp.register")
     def test_register(self, mock_register, mock_finish):
         mock_register.return_value = AquaModel(
             id="test_id",

Original file line number	Diff line number	Diff line change
`@@ -75,7 +75,7 @@ def test_list(self, mock_list):`
`75`	`75`	`)`
`76`	`76`
`77`	`77`	`@patch("notebook.base.handlers.APIHandler.finish")`
`78`		`- @patch("ads.aqua.model.model.AquaModelApp.register")`
	`78`	`+ @patch("ads.aqua.model.AquaModelApp.register")`
`79`	`79`	`def test_register(self, mock_register, mock_finish):`
`80`	`80`	`mock_register.return_value = AquaModel(`
`81`	`81`	`id="test_id",`