grid-cortex-client 0.5.0__tar.gz → 0.5.1__tar.gz

{grid_cortex_client-0.5.0 → grid_cortex_client-0.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: grid-cortex-client
-Version: 0.5.0
+Version: 0.5.1
 Summary: Python client for Grid Cortex
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.8

{grid_cortex_client-0.5.0 → grid_cortex_client-0.5.1}/tests/test_fast_foundation_stereo.py

@@ -18,12 +18,12 @@ class TestFastFoundationStereoHTTP(BaseModelTestHTTP):
     @property
     def batch_size(self) -> int:
         """Fast-FoundationStereo model max_ongoing_requests from config."""
-        return 8
+        return 2
 
     @property
     def gpu_mem(self) -> float:
-        """Fast-FoundationStereo model GPU memory usage in MB."""
-        return 5500.0
+        """Declared GPU memory (MB). Prod pool peaked 7304 MB under max_ongoing=8 (now 2); 7600 = observed + buffer."""
+        return 7600.0
 
     @property
     def max_input_size(self) -> tuple:

{grid_cortex_client-0.5.0 → grid_cortex_client-0.5.1}/tests/test_foundation_stereo.py

@@ -18,12 +18,12 @@ class TestFoundationStereoHTTP(BaseModelTestHTTP):
     @property
     def batch_size(self) -> int:
         """FoundationStereo model max_ongoing_requests from config."""
-        return 8
+        return 2
 
     @property
     def gpu_mem(self) -> float:
-        """FoundationStereo model GPU memory usage in MB."""
-        return 8100.0
+        """Declared GPU memory (MB). Prod pool peaked 9144 MB under max_ongoing=8 (now 2); 9500 = observed + buffer."""
+        return 9500.0
 
     @property
     def max_input_size(self) -> tuple:

{grid_cortex_client-0.5.0 → grid_cortex_client-0.5.1}/tests/test_graspgenx.py

@@ -31,8 +31,8 @@ class TestGraspGenXHTTP(BaseModelTestHTTP):
 
     @property
     def gpu_mem(self) -> float:
-        """Declared GPU memory (MB). ~688 MB peak in CI smoke (num_grasps=16); 1000 = headroom."""
-        return 1000.0
+        """Declared GPU memory (MB). Observed ~2000 MB peak in prod (gripper-dependent / cached samplers); 3000 = buffer."""
+        return 3000.0
 
     @property
     def max_input_size(self) -> tuple: