Fix memory units. (#727)

neuro-inc · Jul 7, 2022 · d19e5d2 · d19e5d2
1 parent 782a4c9
commit d19e5d2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/integration/test_api.py b/tests/integration/test_api.py
@@ -304,7 +304,7 @@ def _factory() -> dict[str, Any]:
             "container": {
                 "image": "ubuntu",
                 "command": "true",
-                "resources": {"cpu": 0.1, "memory_mb": 32},
+                "resources": {"cpu": 0.1, "memory": 32 * 2**20},
             }
         }
 

diff --git a/tests/integration/test_jobs_service.py b/tests/integration/test_jobs_service.py
@@ -229,7 +229,7 @@ async def test_save_pending_job(
         image_tag: str,
     ) -> None:
         resources = Resources(
-            memory=16**10 * 1024**2,
+            memory=2**60,
             cpu=0.1,
             gpu=None,
             shm=False,