responding to mr comments

tableau · Sep 20, 2024 · 90119a1 · 90119a1
1 parent e2761b3
commit 90119a1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tabpy/tabpy_server/handlers/endpoint_handler.py b/tabpy/tabpy_server/handlers/endpoint_handler.py
@@ -73,7 +73,7 @@ def put(self, name):
 
             version_after_update = int(endpoints[name]["version"])
             if request_data.get('should_update_version'):
-                version_after_update = int(endpoints[name]["version"]) + 1
+                version_after_update += 1
             self.logger.log(logging.INFO, f"Endpoint info: {request_data}")
             err_msg = yield self._add_or_update_endpoint(
                 "update", name, version_after_update, request_data

diff --git a/tabpy/tabpy_tools/client.py b/tabpy/tabpy_tools/client.py
@@ -251,7 +251,7 @@ def deploy(self, name, obj, description="", schema=None, override=False, is_publ
         if version == 1:
             self._service.add_endpoint(Endpoint(**obj))
         else:
-            self._service.set_endpoint(Endpoint(**obj))
+            self._service.set_endpoint(Endpoint(**obj), should_update_version=True)
 
         self._wait_for_endpoint_deployment(obj["name"], obj["version"])