Anto01
2b3573ec42
Major additions: - Training data export system for AtomizerField neural network training - Bracket stiffness optimization study with 50+ training samples - Intelligent NX model discovery (auto-detect solutions, expressions, mesh) - Result extractors module for displacement, stress, frequency, mass - User-generated NX journals for advanced workflows - Archive structure for legacy scripts and test outputs - Protocol documentation and dashboard launcher 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
52 lines
2.0 KiB
Python
52 lines
2.0 KiB
Python
"""Reset bracket stiffness optimization study by deleting database and optionally cleaning Nastran files."""
|
|
import optuna
|
|
from pathlib import Path
|
|
import argparse
|
|
|
|
study_dir = Path(__file__).parent
|
|
storage = f"sqlite:///{study_dir / '2_results' / 'study.db'}"
|
|
study_name = "bracket_stiffness_optimization_atomizerfield"
|
|
|
|
def clean_nastran_files():
|
|
"""Remove old Nastran solver output files."""
|
|
model_dir = study_dir / "1_setup" / "model"
|
|
nastran_extensions = ['*.op2', '*.f06', '*.log', '*.f04', '*.pch', '*.DBALL', '*.MASTER', '*.asg', '*.diag']
|
|
temp_patterns = ['_temp*.txt', '*_temp_*']
|
|
|
|
deleted = []
|
|
for pattern in nastran_extensions + temp_patterns:
|
|
for f in model_dir.glob(pattern):
|
|
try:
|
|
f.unlink()
|
|
deleted.append(f.name)
|
|
except Exception as e:
|
|
print(f"[WARNING] Could not delete {f.name}: {e}")
|
|
|
|
return deleted
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Reset bracket optimization study")
|
|
parser.add_argument('--clean', action='store_true', help='Also clean Nastran output files')
|
|
parser.add_argument('--clean-only', action='store_true', help='Only clean Nastran files, keep database')
|
|
args = parser.parse_args()
|
|
|
|
if not args.clean_only:
|
|
try:
|
|
optuna.delete_study(study_name=study_name, storage=storage)
|
|
print(f"[OK] Deleted study: {study_name}")
|
|
except KeyError:
|
|
print(f"[INFO] Study '{study_name}' not found (database may not exist)")
|
|
except Exception as e:
|
|
print(f"[ERROR] Error: {e}")
|
|
|
|
if args.clean or args.clean_only:
|
|
deleted = clean_nastran_files()
|
|
if deleted:
|
|
print(f"[OK] Deleted {len(deleted)} Nastran files:")
|
|
for f in deleted[:5]:
|
|
print(f" - {f}")
|
|
if len(deleted) > 5:
|
|
print(f" ... and {len(deleted) - 5} more")
|
|
else:
|
|
print("[INFO] No Nastran files to clean")
|