Fix tests

Author: jvstme

src/tests/_internal/core/backends/aws/test_resources.py

@@ -118,7 +118,7 @@ def test_returns_the_latest_available(self, ec2_client_mock: Mock):
         }
         image_id, username = get_image_id_and_username(
             ec2_client_mock,
-            cuda=False,
+            gpu_name=None,
             instance_type="some",
         )
         assert image_id == "ami-00000000000000003"
@@ -144,7 +144,7 @@ def test_raises_resource_not_found_if_none_available(
         with pytest.raises(ComputeResourceNotFoundError):
             get_image_id_and_username(
                 ec2_client_mock,
-                cuda=False,
+                gpu_name=None,
                 instance_type="some",
             )
         assert "image 'dstack-0.0' not found" in caplog.text
@@ -162,7 +162,7 @@ def test_uses_dstack_image_name_and_account_id_if_image_config_not_provided(
         monkeypatch.setattr("dstack.version.base_image", "0.0")
         _, username = get_image_id_and_username(
             ec2_client_mock,
-            cuda=cuda,
+            gpu_name="A10G" if cuda else None,
             instance_type="some",
         )
         assert username == "ubuntu"
@@ -198,7 +198,7 @@ def test_uses_image_config_if_provided(
         )
         _, username = get_image_id_and_username(
             ec2_client_mock,
-            cuda=cuda,
+            gpu_name="A10G" if cuda else None,
             instance_type="some",
             image_config=image_config,
         )
@@ -221,7 +221,7 @@ def test_raises_resource_not_found_if_image_config_property_not_set(
         with pytest.raises(ComputeResourceNotFoundError):
             get_image_id_and_username(
                 ec2_client_mock,
-                cuda=False,
+                gpu_name=None,
                 instance_type="some",
                 image_config=image_config,
             )