forgot to remove a comment

mila-iqia · Mar 19, 2024 · 9c7e8ad · 9c7e8ad
1 parent 45b05fe
commit 9c7e8ad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/data/diffusion/test_data_loader.py b/tests/data/diffusion/test_data_loader.py
@@ -1,7 +1,7 @@
 import pytest
 import torch
 
-from crystal_diffusion.data.diffusion.data_loader import LammpsForDiffusionDataModule  # Replace 'your_module' with the actual module name
+from crystal_diffusion.data.diffusion.data_loader import LammpsForDiffusionDataModule
 
 
 @pytest.fixture