Skip to content

Commit 1434c4e

Browse files
committed
fixed unit tests
1 parent 040de20 commit 1434c4e

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/unitary/with_extras/aqua/test_deployment.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2965,7 +2965,9 @@ def setUp(self):
29652965
# Mock internal helpers to avoid real API calls
29662966
self.app.get_container_config = MagicMock()
29672967
self.app.get_container_image = MagicMock(return_value="docker/image:latest")
2968-
self.app.list_shapes = MagicMock(return_value=[MagicMock(name="VM.GPU.A10.1")])
2968+
mock_shape = MagicMock()
2969+
mock_shape.name = "VM.GPU.A10.1"
2970+
self.app.list_shapes = MagicMock(return_value=[mock_shape])
29692971

29702972
# Mock the SMM Defaults (What happens if user sends nothing)
29712973
self.mock_config = MagicMock()

0 commit comments

Comments
 (0)