fix cfg

Signed-off-by: Austin Liu <[email protected]>
austin362667 · May 10, 2024 · 6f2c9ab · 6f2c9ab
1 parent 01d6bae
commit 6f2c9ab
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/flytekit/integration/remote/test_rust_remote.py b/tests/flytekit/integration/remote/test_rust_remote.py
@@ -34,7 +34,7 @@ def test_register_task():
     def my_test_task(n: int) -> int:
         return n
 
-    remote_rs = FlyteRemote(Config.for_endpoint(insecure=True), default_project=PROJECT, default_domain=DOMAIN, enable_rust=True)
+    remote_rs = FlyteRemote(Config.for_endpoint(endpoint="localhost:30080", insecure=True), default_project=PROJECT, default_domain=DOMAIN, enable_rust=True)
     flyte_task = remote_rs.register_task(
         entity=my_test_task,
         serialization_settings=SerializationSettings(
@@ -51,7 +51,7 @@ def test_fetch_task_without_grpc():
     # @task()
     # def my_test_task(n: int) -> int:
     #     return n
-    remote_rs = FlyteRemote(Config.for_endpoint(insecure=True), default_project=PROJECT, default_domain=DOMAIN, enable_rust=True)
+    remote_rs = FlyteRemote(Config.for_endpoint(endpoint="localhost:30080", insecure=True), default_project=PROJECT, default_domain=DOMAIN, enable_rust=True)
     # remote_rs.register_task(
     #     entity=my_test_task,
     #     serialization_settings=SerializationSettings(
@@ -68,11 +68,11 @@ def test_fetch_task_without_grpc():
 @pytest.mark.skipif(not check_python_installation("flyrs"), reason="Rust client is not installed.")
 @pytest.mark.skipif(not check_python_installation("grpc"), reason="gRPC is not installed.")
 def test_fetch_task_and_compare():
-    remote_py = FlyteRemote(Config.for_endpoint(insecure=True), default_project=PROJECT, default_domain=DOMAIN)
+    remote_py = FlyteRemote(Config.for_endpoint(endpoint="localhost:30080", insecure=True), default_project=PROJECT, default_domain=DOMAIN)
     # @task()
     # def my_test_task(n: int) -> int:
     #     return n
-    remote_rs = FlyteRemote(Config.for_endpoint(insecure=True), default_project=PROJECT, default_domain=DOMAIN, enable_rust=True)
+    remote_rs = FlyteRemote(Config.for_endpoint(endpoint="localhost:30080", insecure=True), default_project=PROJECT, default_domain=DOMAIN, enable_rust=True)
     # remote_rs.register_task(
     #     entity=my_test_task,
     #     serialization_settings=SerializationSettings(