fix customized CV files

deepmodeling · Aug 29, 2023 · 40d491c · 40d491c
1 parent d6f2d70
commit 40d491c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rid/op/run_label.py b/rid/op/run_label.py
@@ -146,7 +146,7 @@ def execute(
                     os.symlink(op_in["index_file"], op_in["index_file"].name)
             if op_in["cv_file"] is not None:
                 for file in op_in["cv_file"]:
-                    if file.name != "colvar":
+                    if os.path.basename(file).endswith("pdb"):
                         if not os.path.islink(file.name):
                             os.symlink(file, file.name)
             if op_in["inputfile"] is not None: