matrice-compute 0.1.27__py3-none-any.whl → 0.1.28__py3-none-any.whl

matrice_compute/instance_manager.py

@@ -41,6 +41,7 @@ class InstanceManager:
         env: str = "",
         gpus: str = "",
         workspace_dir: str = "matrice_workspace",
+        enable_kafka: bool = False,
     ):
         """Initialize an instance manager.
 
@@ -61,6 +62,7 @@ class InstanceManager:
                 Defaults to empty string.
             workspace_dir (str): Directory for workspace files.
                 Defaults to "matrice_workspace".
+            enable_kafka (bool): Enable Kafka communication (default False).
         """
         self.session = self._setup_env_credentials(
             env,
@@ -75,6 +77,7 @@ class InstanceManager:
         self.scaling = Scaling(
             self.session,
             os.environ.get("INSTANCE_ID"),
+            enable_kafka,
         )
         logging.info("InstanceManager initialized with scaling")
         jupyter_token = os.environ.get("JUPYTER_TOKEN")

matrice_compute/scaling.py

@@ -18,7 +18,7 @@ class Scaling:
 
     """Class providing scaling functionality for compute instances."""
 
-    def __init__(self, session, instance_id=None, enable_kafka=True):
+    def __init__(self, session, instance_id=None, enable_kafka=False):
         """Initialize Scaling instance.
 
         Args:

{matrice_compute-0.1.27.dist-info → matrice_compute-0.1.28.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matrice_compute
-Version: 0.1.27
+Version: 0.1.28
 Summary: Common server utilities for Matrice.ai services
 Author-email: "Matrice.ai" <dipendra@matrice.ai>
 License-Expression: MIT

{matrice_compute-0.1.27.dist-info → matrice_compute-0.1.28.dist-info}/RECORD

@@ -3,16 +3,16 @@ matrice_compute/action_instance.py,sha256=upGZaluoVawusKFmULfk27hHJmfRi23m5W-BNV
 matrice_compute/actions_manager.py,sha256=Iex5uw0PLRR4pvIAZDxc2CypucbanKDbJ3SK8mMGXK8,18148
 matrice_compute/actions_scaledown_manager.py,sha256=pJ0nduNwHWZ10GnqJNx0Ok7cVWabQ_M8E2Vb9pH3A_k,2002
 matrice_compute/compute_operations_handler.py,sha256=amcMhmXtv2irE6qK8Vbgec_8uFqjWmVVp0VWq-73_MU,17781
-matrice_compute/instance_manager.py,sha256=4AMOsA3eazb7qBuVRZDyF3eTzGVXKtKpFQBcSF5IWLg,19153
+matrice_compute/instance_manager.py,sha256=kPZYfiq3Oevs5r1xzwvDzE27zeWF9oBBxh9KhpHJuG4,19292
 matrice_compute/instance_utils.py,sha256=N4yPDvNukFEEBngR0lEt4x_XT5hur1q0P-spM2xQIlU,42025
 matrice_compute/prechecks.py,sha256=W9YmNF3RcLhOf4U8WBlExvFqDw1aGWSNTlJtA73lbDQ,17196
 matrice_compute/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 matrice_compute/resources_tracker.py,sha256=1jSLrIFlOh-vgyNzFrUrE2Ak2JAGCIfV7wcyEPJ0f2c,32246
-matrice_compute/scaling.py,sha256=o-lan8RFC3NSUpaFtz37vytYiD5vxATHULIUwtH2uXA,52930
+matrice_compute/scaling.py,sha256=oeCgQb6g9W7p8-VaDXT4jDH3aMz-fCO5MZmj-1Gbz1c,52931
 matrice_compute/shutdown_manager.py,sha256=0MYV_AqygqR9NEntYf7atUC-PbWXyNkm1f-8c2aizgA,13234
 matrice_compute/task_utils.py,sha256=3qIutiQdYPyGRxH9ZwLbqdg8sZcnp6jp08pszWCRFl0,2820
-matrice_compute-0.1.27.dist-info/licenses/LICENSE.txt,sha256=_uQUZpgO0mRYL5-fPoEvLSbNnLPv6OmbeEDCHXhK6Qc,1066
-matrice_compute-0.1.27.dist-info/METADATA,sha256=fp05SYBiLtgtyJlysW9CaL8lCgrQbfTpjmOEyYcRyu4,1038
-matrice_compute-0.1.27.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-matrice_compute-0.1.27.dist-info/top_level.txt,sha256=63Plr3L1GzBUWZO5JZaFkiv8IcB10xUPU-9w3i6ptvE,16
-matrice_compute-0.1.27.dist-info/RECORD,,
+matrice_compute-0.1.28.dist-info/licenses/LICENSE.txt,sha256=_uQUZpgO0mRYL5-fPoEvLSbNnLPv6OmbeEDCHXhK6Qc,1066
+matrice_compute-0.1.28.dist-info/METADATA,sha256=1kig3ZTOXrgX71W2FtcgyZ9RO_xyUq8b1_qYrl0qhpA,1038
+matrice_compute-0.1.28.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+matrice_compute-0.1.28.dist-info/top_level.txt,sha256=63Plr3L1GzBUWZO5JZaFkiv8IcB10xUPU-9w3i6ptvE,16
+matrice_compute-0.1.28.dist-info/RECORD,,