chore: Fixed the architecture/test_vertexai_import test under Pytho…

…n 3.12 when using editable install Fixes: #4037 PiperOrigin-RevId: 649143541
googleapis · Jul 3, 2024 · a6f68df · a6f68df
1 parent 02ffd27
commit a6f68df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/architecture/test_vertexai_import.py b/tests/unit/architecture/test_vertexai_import.py
@@ -34,7 +34,7 @@ def test_vertexai_import():
     # My solution is to ignore modules that are parents of the current test module.
     modules_before_aip = set(sys.modules)
     for module_name in modules_before_aip:
-        assert "aiplatform" not in module_name or __name__.startswith(module_name)
+        assert ".aiplatform" not in module_name or __name__.startswith(module_name)
         assert "vertexai" not in module_name or __name__.startswith(module_name)
 
     # Test aiplatform import