fix: put traversal_paths in kwargs

jina-ai · Aug 2, 2022 · b753673 · b753673
1 parent 56f2289
commit b753673
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 7 deletions.
diff --git a/server/clip_server/executors/clip_onnx.py b/server/clip_server/executors/clip_onnx.py
@@ -24,19 +24,18 @@ def __init__(
         num_worker_preprocess: int = 4,
         minibatch_size: int = 32,
         access_paths: str = '@r',
-        traversal_paths: Optional[str] = '@r',
         model_path: Optional[str] = None,
         **kwargs,
     ):
         super().__init__(**kwargs)
 
         self._minibatch_size = minibatch_size
         self._access_paths = access_paths
-        if traversal_paths is not None:
+        if 'traversal_paths' in kwargs:
             warnings.warn(
                 f'`traversal_paths` is deprecated. Use `access_paths` instead.'
             )
-            self._access_paths = traversal_paths
+            self._access_paths = kwargs['traversal_paths']
 
         self._pool = ThreadPool(processes=num_worker_preprocess)
 

diff --git a/server/clip_server/executors/clip_tensorrt.py b/server/clip_server/executors/clip_tensorrt.py
@@ -32,11 +32,11 @@ def __init__(
 
         self._minibatch_size = minibatch_size
         self._access_paths = access_paths
-        if traversal_paths is not None:
+        if 'traversal_paths' in kwargs:
             warnings.warn(
                 f'`traversal_paths` is deprecated. Use `access_paths` instead.'
             )
-            self._access_paths = traversal_paths
+            self._access_paths = kwargs['traversal_paths']
 
         self._device = device
 

diff --git a/server/clip_server/executors/clip_torch.py b/server/clip_server/executors/clip_torch.py
@@ -33,11 +33,11 @@ def __init__(
 
         self._minibatch_size = minibatch_size
         self._access_paths = access_paths
-        if traversal_paths is not None:
+        if 'traversal_paths' in kwargs:
             warnings.warn(
                 f'`traversal_paths` is deprecated. Use `access_paths` instead.'
             )
-            self._access_paths = traversal_paths
+            self._access_paths = kwargs['traversal_paths']
 
         if not device:
             self._device = 'cuda' if torch.cuda.is_available() else 'cpu'