fix slicing error

deepmodeling · Oct 23, 2023 · ef61957 · ef61957
1 parent 9bcb215
commit ef61957
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepmd/entrypoints/convert.py b/deepmd/entrypoints/convert.py
@@ -18,8 +18,8 @@ def convert(
     output_model: str,
     **kwargs,
 ):
-    if output_model[:-6] == ".pbtxt":
-        if input_model[:-6] != ".pbtxt":
+    if output_model[-6:] == ".pbtxt":
+        if input_model[-6:] != ".pbtxt":
             convert_pb_to_pbtxt(input_model, output_model)
         else:
             raise RuntimeError("input model is already pbtx")