Upload folder using huggingface_hub

bhashini.py

@@ -6,46 +6,56 @@ import threading
 import sys
 import types
 
-# --- CRITICAL SPRINT 18 PATCH ---
-# IndicTrans2's configuration_indictrans.py imports from `transformers.onnx` and
-# `transformers.onnx.utils`, which were removed in transformers>=4.40.
-# We mock the ENTIRE onnx subpackage tree as a proper package (with __path__)
-# so Python's import resolver can resolve submodule paths like `transformers.onnx.utils`.
-# This allows compatibility with transformers>=4.48.0 (required by Ettin Reranker/ModernBERT).
+# --- CRITICAL SPRINT 18 PATCH (DEFINITIVE FIX) ---
+# IndicTrans2's remote `configuration_indictrans.py` imports many classes and functions
+# from `transformers.onnx` (OnnxConfig, OnnxSeq2SeqConfigWithPast, compute_effective_axis_dimension, etc.)
+# This subpackage was REMOVED in transformers>=4.40, but we need transformers>=4.48 for ModernBERT.
+#
+# SOLUTION: Universal mock modules that return a no-op dummy for ANY attribute access.
+# The ONNX classes are ONLY used for ONNX model export — never during inference.
+# So returning harmless dummies is 100% safe for our translation pipeline.
 if 'transformers.onnx' not in sys.modules:
-    # Base mock classes that IndicTrans2 config tries to inherit from
-    class MockOnnxConfig:
-        pass
-    class MockOnnxSeq2SeqConfigWithPast(MockOnnxConfig):
-        pass
 
-    # Create transformers.onnx as a PACKAGE (needs __path__ for submodule resolution)
-    mock_onnx = types.ModuleType('transformers.onnx')
-    mock_onnx.__path__ = []  # Makes Python treat it as a package, not a plain module
+    class _UniversalDummy:
+        """A universal no-op class that acts as a stand-in for any ONNX class/function.
+        Can be instantiated, called, subclassed, and used as a decorator — all no-ops."""
+        def __init__(self, *a, **kw): pass
+        def __call__(self, *a, **kw): return self
+        def __getattr__(self, name): return _UniversalDummy()
+        def __init_subclass__(cls, **kw): pass
+
+    class _MockOnnxModule(types.ModuleType):
+        """A module that returns _UniversalDummy for any attribute access.
+        This means `from transformers.onnx.utils import <anything>` always succeeds."""
+        def __getattr__(self, name):
+            return _UniversalDummy
+
+    # Build the mock package tree
+    mock_onnx = _MockOnnxModule('transformers.onnx')
+    mock_onnx.__path__ = []           # Makes Python treat it as a package
     mock_onnx.__package__ = 'transformers.onnx'
-    mock_onnx.OnnxConfig = MockOnnxConfig
-    mock_onnx.OnnxSeq2SeqConfigWithPast = MockOnnxSeq2SeqConfigWithPast
 
-    # Create transformers.onnx.utils submodule
-    mock_onnx_utils = types.ModuleType('transformers.onnx.utils')
+    mock_onnx_utils = _MockOnnxModule('transformers.onnx.utils')
     mock_onnx_utils.__package__ = 'transformers.onnx'
 
-    # Create transformers.onnx.config submodule (some model configs import from here)
-    mock_onnx_config = types.ModuleType('transformers.onnx.config')
+    mock_onnx_config = _MockOnnxModule('transformers.onnx.config')
     mock_onnx_config.__package__ = 'transformers.onnx'
-    mock_onnx_config.OnnxConfig = MockOnnxConfig
-    mock_onnx_config.OnnxSeq2SeqConfigWithPast = MockOnnxSeq2SeqConfigWithPast
 
-    # Register all mock modules in sys.modules
+    mock_onnx_features = _MockOnnxModule('transformers.onnx.features')
+    mock_onnx_features.__package__ = 'transformers.onnx'
+
+    # Register in sys.modules (covers all possible import paths)
     sys.modules['transformers.onnx'] = mock_onnx
     sys.modules['transformers.onnx.utils'] = mock_onnx_utils
     sys.modules['transformers.onnx.config'] = mock_onnx_config
+    sys.modules['transformers.onnx.features'] = mock_onnx_features
 
-    # Wire submodules as attributes on parent (Python import system requires this)
+    # Wire submodules as attributes on parent
     mock_onnx.utils = mock_onnx_utils
     mock_onnx.config = mock_onnx_config
+    mock_onnx.features = mock_onnx_features
 
-    # Wire onnx onto the real transformers package
+    # Wire onto the real transformers package
     import transformers
     transformers.onnx = mock_onnx
 # --------------------------------