kfoughali commited on
Commit
e4e8b6b
·
verified ·
1 Parent(s): 991b7c0

Update core/graph_mamba.py

Browse files
Files changed (1) hide show
  1. core/graph_mamba.py +3 -3
core/graph_mamba.py CHANGED
@@ -350,9 +350,9 @@ def create_astrocyte_config():
350
  'input_dim': 1433
351
  }
352
 
353
- # Aliases
354
- AstrocyteGraphMamba = CognitiveMambaGraphMamba
355
- GraphMamba = CognitiveMambaGraphMamba
356
  HybridGraphMamba = LegacyGraphMamba
357
  QuantumEnhancedGraphMamba = LegacyGraphMamba
358
  create_regularized_config = create_astrocyte_config
 
350
  'input_dim': 1433
351
  }
352
 
353
+ # Use simple working version for now
354
+ AstrocyteGraphMamba = LegacyGraphMamba
355
+ GraphMamba = LegacyGraphMamba
356
  HybridGraphMamba = LegacyGraphMamba
357
  QuantumEnhancedGraphMamba = LegacyGraphMamba
358
  create_regularized_config = create_astrocyte_config