Fix ModuleSpec constructor by removing unsupported 'parent' parameter
Browse files
app.py
CHANGED
|
@@ -104,19 +104,18 @@ def create_in_memory_flash_attn_mock():
|
|
| 104 |
# Add attributes used by transformers checks
|
| 105 |
flash_attn.__version__ = "0.0.0-mocked"
|
| 106 |
|
| 107 |
-
# Create common submodules
|
| 108 |
for submodule in ['flash_attn.flash_attn_interface', 'flash_attn.flash_attn_triton']:
|
| 109 |
parts = submodule.split('.')
|
| 110 |
parent_name = '.'.join(parts[:-1])
|
| 111 |
child_name = parts[-1]
|
| 112 |
parent = sys.modules[parent_name]
|
| 113 |
|
| 114 |
-
# Create submodule spec
|
| 115 |
subspec = importlib.machinery.ModuleSpec(
|
| 116 |
name=submodule,
|
| 117 |
loader=DummyLoader(submodule),
|
| 118 |
-
is_package=False
|
| 119 |
-
parent=parent
|
| 120 |
)
|
| 121 |
|
| 122 |
# Create and register submodule
|
|
|
|
| 104 |
# Add attributes used by transformers checks
|
| 105 |
flash_attn.__version__ = "0.0.0-mocked"
|
| 106 |
|
| 107 |
+
# Create common submodules - FIXED to not use 'parent' parameter in ModuleSpec
|
| 108 |
for submodule in ['flash_attn.flash_attn_interface', 'flash_attn.flash_attn_triton']:
|
| 109 |
parts = submodule.split('.')
|
| 110 |
parent_name = '.'.join(parts[:-1])
|
| 111 |
child_name = parts[-1]
|
| 112 |
parent = sys.modules[parent_name]
|
| 113 |
|
| 114 |
+
# Create submodule spec - removed 'parent' parameter
|
| 115 |
subspec = importlib.machinery.ModuleSpec(
|
| 116 |
name=submodule,
|
| 117 |
loader=DummyLoader(submodule),
|
| 118 |
+
is_package=False
|
|
|
|
| 119 |
)
|
| 120 |
|
| 121 |
# Create and register submodule
|