Add pat arg to input obj

Clarifai · Nov 23, 2023 · 5e495e9 · 5e495e9
1 parent 2371690
commit 5e495e9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/clarifai/client/dataset.py b/clarifai/client/dataset.py
@@ -64,7 +64,7 @@ def __init__(self,
     self.max_retires = 10
     self.batch_size = 128  # limit max protos in a req
     self.task = None  # Upload dataset type
-    self.input_object = Inputs(user_id=self.user_id, app_id=self.app_id)
+    self.input_object = Inputs(user_id=self.user_id, app_id=self.app_id, pat=pat)
     self.logger = get_logger(logger_level="INFO")
     BaseClient.__init__(self, user_id=self.user_id, app_id=self.app_id, base=base_url, pat=pat)
     Lister.__init__(self)

diff --git a/clarifai/client/search.py b/clarifai/client/search.py
@@ -46,7 +46,7 @@ def __init__(self,
     self.data_proto = resources_pb2.Data()
     self.top_k = top_k
 
-    self.inputs = Inputs(user_id=self.user_id, app_id=self.app_id)
+    self.inputs = Inputs(user_id=self.user_id, app_id=self.app_id, pat=pat)
     self.rank_filter_schema = get_schema()
     BaseClient.__init__(self, user_id=self.user_id, app_id=self.app_id, base=base_url, pat=pat)
     Lister.__init__(self, page_size=1000)