Update setup.py

setup.py

@@ -71,7 +71,12 @@ def get_version_add(sha: Optional[str] = None) -> str:
     with open(add_version_path, 'r',encoding='utf-8') as file:
         lines = file.readlines()
         
+    if len(lines) > 1:
         lines[1] = "__dcu_version__ = '0.0.6+{}'\n".format(version)
+    else:
+        # 如果文件行数不足，可以选择添加空行或者抛出异常
+        lines.append("__dcu_version__ = '0.0.6+{}'\n".format(version))
+        print("Added missing line to the file content.")
     with open(add_version_path, encoding="utf-8",mode="w") as file:
         file.writelines(lines)
     file.close()