mknolan commited on
Commit
dd538c2
·
verified ·
1 Parent(s): 79f6e49

Fix ModuleSpec constructor by removing unsupported 'parent' parameter

Browse files
Files changed (1) hide show
  1. app.py +3 -4
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