[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
@@ -399,7 +399,15 @@ def main(yaml_file, target_node=None, test_mode="oob", clean_up=True):
|
||||
|
||||
# Delete labels for current node configuration
|
||||
print(f"Deleting labels for {node} nodes...")
|
||||
cmd = [python_cmd, "deploy.py", "--chart-name", chart_name, "--num-nodes", str(node), "--delete-label"]
|
||||
cmd = [
|
||||
python_cmd,
|
||||
"deploy.py",
|
||||
"--chart-name",
|
||||
chart_name,
|
||||
"--num-nodes",
|
||||
str(node),
|
||||
"--delete-label",
|
||||
]
|
||||
if current_node_names:
|
||||
cmd.extend(["--node-names"] + current_node_names)
|
||||
|
||||
@@ -410,7 +418,7 @@ def main(yaml_file, target_node=None, test_mode="oob", clean_up=True):
|
||||
except Exception as e:
|
||||
print(f"Error while deleting labels for {node} nodes: {str(e)}")
|
||||
else:
|
||||
print(f"Skipping cleanup for local debug. Manual cleanup may be required.")
|
||||
print("Skipping cleanup for local debug. Manual cleanup may be required.")
|
||||
exit(0)
|
||||
|
||||
except Exception as e:
|
||||
@@ -429,7 +437,9 @@ if __name__ == "__main__":
|
||||
parser.add_argument("yaml_file", help="Path to the YAML configuration file")
|
||||
parser.add_argument("--target-node", type=int, help="Optional: Target number of nodes to deploy.", default=None)
|
||||
parser.add_argument("--test-mode", type=str, help="Test mode, either 'oob' (out of box) or 'tune'.", default="oob")
|
||||
parser.add_argument("--clean-up", type=bool, help="Clean up after test, which can be closed for local debug.", default=True)
|
||||
parser.add_argument(
|
||||
"--clean-up", type=bool, help="Clean up after test, which can be closed for local debug.", default=True
|
||||
)
|
||||
|
||||
args = parser.parse_args()
|
||||
main(args.yaml_file, args.target_node, args.test_mode, args.clean_up)
|
||||
|
||||
Reference in New Issue
Block a user