clarifai 11.1.7rc4__py3-none-any.whl → 11.1.7rc5__py3-none-any.whl

clarifai/__init__.py

@@ -1 +1 @@
-__version__ = "11.1.7rc4"
+__version__ = "11.1.7rc5"

clarifai/runners/models/model_builder.py

@@ -68,11 +68,11 @@ class ModelBuilder:
     self.inference_compute_info = self._get_inference_compute_info()
     self.is_v3 = True  # Do model build for v3
 
-  def create_model_instance(self, load_model=True):
+  def create_model_instance(self, load_model=True, mocking=False):
     """
     Create an instance of the model class, as specified in the config file.
     """
-    model_class = self.load_model_class()
+    model_class = self.load_model_class(mocking=mocking)
 
     # initialize the model
     model = model_class()
@@ -80,7 +80,7 @@ class ModelBuilder:
       model.load_model()
     return model
 
-  def load_model_class(self):
+  def load_model_class(self, mocking=False):
     """
     Import the model class from the model.py file, dynamically handling missing dependencies
     """
@@ -108,9 +108,10 @@ class ModelBuilder:
 
       # Mock all third-party imports to avoid ImportErrors or other issues
       return MagicMock()
-
-    # Replace the built-in __import__ function with our custom one
-    builtins.__import__ = custom_import
+    
+    if mocking:
+      # Replace the built-in __import__ function with our custom one
+      builtins.__import__ = custom_import
 
     try:
       spec.loader.exec_module(module)
@@ -306,7 +307,7 @@ class ModelBuilder:
     """
     Returns the method signatures for the model class in YAML format.
     """
-    model_class = self.load_model_class()
+    model_class = self.load_model_class(mocking=True)
     method_info = model_class._get_method_info()
     signatures = {method.name: method.signature for method in method_info.values()}
     return signatures_to_yaml(signatures)
@@ -315,7 +316,7 @@ class ModelBuilder:
     """
     Returns the method signatures for the model class.
     """
-    model_class = self.load_model_class()
+    model_class = self.load_model_class(mocking=True)
     method_info = model_class._get_method_info()
     signatures = [method.signature for method in method_info.values()]
     return signatures

{clarifai-11.1.7rc4.dist-info → clarifai-11.1.7rc5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: clarifai
-Version: 11.1.7rc4
+Version: 11.1.7rc5
 Summary: Clarifai Python SDK
 Home-page: https://github.com/Clarifai/clarifai-python
 Author: Clarifai

{clarifai-11.1.7rc4.dist-info → clarifai-11.1.7rc5.dist-info}/RECORD

@@ -1,4 +1,4 @@
-clarifai/__init__.py,sha256=ifkdBMaRvYCzXalTvOKcl2DNqYnmG2YMz8JBP1t1moU,26
+clarifai/__init__.py,sha256=L7ecgzwVF1qIM4sX3JzNrp3tI1YSfegldgbqVv-b-HQ,26
 clarifai/cli.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clarifai/errors.py,sha256=RwzTajwds51wLD0MVlMC5kcpBnzRpreDLlazPSBZxrg,2605
 clarifai/versions.py,sha256=jctnczzfGk_S3EnVqb2FjRKfSREkNmvNEwAAa_VoKiQ,222
@@ -138,7 +138,7 @@ clarifai/runners/dockerfile_template/Dockerfile.nim,sha256=CSdUAehj3uOwminioLnT5
 clarifai/runners/dockerfile_template/Dockerfile.template,sha256=5cjv7U8PmWa3DB_5B1CqSYh_6GE0E0np52TIAa7EIDE,2312
 clarifai/runners/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 clarifai/runners/models/base_typed_model.py,sha256=0QCWxch8CcyJSKvE1D4PILd2RSnQZHTmx4DXlQQ6dpo,7856
-clarifai/runners/models/model_builder.py,sha256=GlMpJn1krpckvI0cqd8ujrCnoU62z-kqSI3nJfGJEwY,34788
+clarifai/runners/models/model_builder.py,sha256=DDE4s1yzomsGyl4PbpjKdiofM6VhL0sgZujCP6wvF2k,34881
 clarifai/runners/models/model_class.py,sha256=qK5qk4geoFpqrRtl4VmKK90CIJKyWmD_vjJMqLWR_CQ,11854
 clarifai/runners/models/model_class_refract.py,sha256=HxuozxSW7ag5yWCPxjNwgLArQ6dORhyGXlnpPaZz2-c,3211
 clarifai/runners/models/model_run_locally.py,sha256=VZetm9Mko8MBjcjwr6PCnTU9gF3glgD5qvpbj-8tW2s,17962
@@ -229,9 +229,9 @@ clarifai/workflows/__pycache__/__init__.cpython-39.pyc,sha256=9nA--jULSW7OFrYOcs
 clarifai/workflows/__pycache__/export.cpython-310.pyc,sha256=phEGwi2gAojCUhRTqjZVeTDn7Gk6LCVBeSTjAj4m9iY,2418
 clarifai/workflows/__pycache__/utils.cpython-310.pyc,sha256=M9_KTM7GOOS5SPrWwAzqHDqyGvgKi3xuSGvyw6MNf-I,1925
 clarifai/workflows/__pycache__/validate.cpython-310.pyc,sha256=c18Jgp_-CAm8RD_tmUpDCPoqZeexaoWELG0yBzb9rjw,2149
-clarifai-11.1.7rc4.dist-info/LICENSE,sha256=mUqF_d12-qE2n41g7C5_sq-BMLOcj6CNN-jevr15YHU,555
-clarifai-11.1.7rc4.dist-info/METADATA,sha256=kufmH0KOUsaV0LTIz5ckGf8oF4zfCr5giQbrsAucNRo,22215
-clarifai-11.1.7rc4.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-clarifai-11.1.7rc4.dist-info/entry_points.txt,sha256=X9FZ4Z-i_r2Ud1RpZ9sNIFYuu_-9fogzCMCRUD9hyX0,51
-clarifai-11.1.7rc4.dist-info/top_level.txt,sha256=wUMdCQGjkxaynZ6nZ9FAnvBUCgp5RJUVFSy2j-KYo0s,9
-clarifai-11.1.7rc4.dist-info/RECORD,,
+clarifai-11.1.7rc5.dist-info/LICENSE,sha256=mUqF_d12-qE2n41g7C5_sq-BMLOcj6CNN-jevr15YHU,555
+clarifai-11.1.7rc5.dist-info/METADATA,sha256=tOQbvNsbo_DjwJRDarrhk1uAgnSjSCDfdlcaUucYhI8,22215
+clarifai-11.1.7rc5.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+clarifai-11.1.7rc5.dist-info/entry_points.txt,sha256=X9FZ4Z-i_r2Ud1RpZ9sNIFYuu_-9fogzCMCRUD9hyX0,51
+clarifai-11.1.7rc5.dist-info/top_level.txt,sha256=wUMdCQGjkxaynZ6nZ9FAnvBUCgp5RJUVFSy2j-KYo0s,9
+clarifai-11.1.7rc5.dist-info/RECORD,,