(test)

tests/manager/conftest.py

@@ -67,10 +67,12 @@
     vfolders,
 )
 from ai.backend.manager.models.base import (
-    load_table,
+    metadata,
     pgsql_connect_opts,
     populate_fixture,
 )
+
+# from ai.backend.manager.models.container_registry import ContainerRegistry
 from ai.backend.manager.models.scaling_group import ScalingGroupOpts
 from ai.backend.manager.models.utils import connect_database
 from ai.backend.manager.registry import AgentRegistry
@@ -415,6 +417,7 @@ def database_fixture(local_config, test_db, database):
     Populate the example data as fixtures to the database
     and delete them after use.
     """
+    print("database_fixture!!!ewodkpw")
     db_addr = local_config["db"]["addr"]
     db_user = local_config["db"]["user"]
     db_pass = local_config["db"]["password"]
@@ -464,7 +467,8 @@ async def clean_fixture():
                 await conn.execute((users.delete()))
                 await conn.execute((scaling_groups.delete()))
                 await conn.execute((domains.delete()))
-                await conn.execute(((await load_table(engine, "container_registries")).delete()))
+                # await conn.execute(((await load_table(engine, "container_registries")).delete()))
+                await conn.execute((metadata.tables.get("container_registries")))
         finally:
             await engine.dispose()