fix: enable autoscaling for websocket flows

jina-ai · Apr 20, 2023 · b84f5d2 · b84f5d2
1 parent d27e305
commit b84f5d2
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/lcserve/flow.py b/lcserve/flow.py
@@ -381,12 +381,6 @@ def get_gateway_jcloud_args(
     autoscale: AutoscaleConfig = AutoscaleConfig(),
     websocket: bool = False,
 ) -> Dict:
-    _autoscale_args = autoscale.to_dict() if autoscale else {}
-    if (
-        websocket
-    ):  # # TODO: remove this when websocket + autoscale is supported in JCloud
-        _autoscale_args = {}
-
     return {
         'jcloud': {
             'expose': True,
@@ -395,7 +389,7 @@ def get_gateway_jcloud_args(
                 'capacity': 'spot',
             },
             'healthcheck': False if websocket else True,
-            **_autoscale_args,
+            **(autoscale.to_dict() if autoscale else {}),
         }
     }