Files
Atomizer/studies/uav_arm_atomizerfield_test/reset_study.py
Anto01 a4805947d1 feat: Add NX study models and optimization histories
Includes all study folders with NX models for development:
- bracket_stiffness_optimization (V1, V2, V3)
- drone_gimbal_arm_optimization
- simple_beam_optimization
- uav_arm_optimization (V1, V2)
- training_data_export_test
- uav_arm_atomizerfield_test

Contains .prt, .fem, .sim files and optimization databases.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
2025-11-26 12:19:07 -05:00

38 lines
1.0 KiB
Python

"""
Reset the UAV arm AtomizerField test study
Clears all optimization results and logs to start fresh.
Preserves the setup and model files.
"""
from pathlib import Path
import shutil
def reset_study():
"""Reset study to clean state."""
study_dir = Path(__file__).parent
# Remove results directory
results_dir = study_dir / "2_results"
if results_dir.exists():
print(f"Removing {results_dir}...")
shutil.rmtree(results_dir)
# Remove training data export directory if exists
training_data_dir = Path("atomizer_field_training_data/uav_arm_test")
if training_data_dir.exists():
print(f"Removing {training_data_dir}...")
shutil.rmtree(training_data_dir)
# Remove any Python cache
pycache = study_dir / "__pycache__"
if pycache.exists():
print(f"Removing {pycache}...")
shutil.rmtree(pycache)
print("Study reset complete!")
print("You can now run: python run_optimization.py")
if __name__ == "__main__":
reset_study()