File size: 879 Bytes
b47a1ce | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
from pathlib import Path
def test_new_standalone_files_do_not_reference_old_ssm_method_paths():
forbidden = [
"spatial_ssm_memory",
"df_video_ssm_memory",
"StateSpaceSpatialMemoryMinecraft",
"ssm_memory_ckpt_path",
"dememwm_compression",
]
for rel in ["algorithms/worldmem/dememwm_memory_dit.py", "algorithms/worldmem/dememwm/algorithm.py", "configurations/algorithm/dememwm_memory_dit.yaml"]:
text = Path(rel).read_text()
for token in forbidden:
assert token not in text, f"{token} leaked into {rel}"
def test_legacy_ssm_bridge_scaffold_is_not_present():
assert not Path("algorithms/worldmem/models/dememwm_compression.py").exists()
assert not Path("algorithms/worldmem/df_video_ssm_memory.py").exists()
assert not Path("algorithms/worldmem/models/spatial_ssm_memory.py").exists()
|