drbx 2.0.0.dev0__py3-none-any.whl

drbx/validation/__init__.py

@@ -0,0 +1,424 @@
+from .autodiff_diffusion import (
+    DiffusionAutodiffSetup,
+    StrongScalingPoint,
+    build_diffusion_autodiff_setup,
+    compute_strong_scaling_points,
+    design_field,
+    design_field_from_physical,
+    finite_difference_gradient,
+    objective_for_parameter_vector,
+    objective_for_physical_parameters,
+    optimize_inverse_design,
+    physical_to_theta,
+    simulate_density_history,
+    simulate_density_history_from_physical,
+    theta_to_physical,
+)
+from .autodiff_diffusion_uncertainty import (
+    AutodiffDiffusionUncertaintyArtifacts,
+    build_autodiff_diffusion_uncertainty_report,
+    create_autodiff_diffusion_uncertainty_package,
+    save_autodiff_diffusion_uncertainty_plot,
+)
+from .diverted_tokamak_movie import (
+    DivertedTokamakFieldHistory,
+    DivertedTokamakGeometry,
+    DivertedTokamakMovieArtifacts,
+    assemble_tokamak_rank_history,
+    build_diverted_tokamak_analysis,
+    create_diverted_tokamak_movie_package,
+    create_diverted_tokamak_movie_package_from_arrays,
+    load_diverted_tokamak_arrays_npz,
+    load_diverted_tokamak_geometry,
+    save_diverted_tokamak_gif,
+    save_diverted_tokamak_poster_frame,
+    save_diverted_tokamak_snapshot_panel,
+    toroidal_mean_fluctuation,
+    write_diverted_tokamak_analysis_json,
+    write_diverted_tokamak_arrays_npz,
+)
+from .essos_fieldline_import_campaign import (
+    EssosFieldLineImportArtifacts,
+    build_essos_fieldline_import_report,
+    create_essos_fieldline_import_package,
+    save_essos_fieldline_import_plot,
+)
+from .essos_vmec_fieldline_surface_campaign import (
+    EssosVmecFieldlineSurfaceArtifacts,
+    build_essos_vmec_fieldline_surface_campaign,
+    create_essos_vmec_fieldline_surface_package,
+    save_essos_vmec_fieldline_surface_plot,
+)
+from .essos_vmec_closed_field_campaign import (
+    EssosVmecClosedFieldArtifacts,
+    EssosVmecClosedFieldDryRunArtifacts,
+    build_essos_vmec_closed_field_report,
+    create_essos_vmec_closed_field_dry_run_package,
+    create_essos_vmec_closed_field_package,
+    save_essos_vmec_closed_field_plot,
+)
+from .essos_vmec_closed_field_transient_campaign import (
+    EssosVmecClosedFieldTransientArtifacts,
+    EssosVmecClosedFieldTransientDryRunArtifacts,
+    build_essos_vmec_closed_field_transient_campaign,
+    create_essos_vmec_closed_field_transient_dry_run_package,
+    create_essos_vmec_closed_field_transient_package,
+    create_essos_vmec_closed_field_transient_package_from_geometry,
+    save_essos_vmec_closed_field_transient_movie,
+    save_essos_vmec_closed_field_transient_plot,
+)
+from .vmec_extender_edge_field_campaign import (
+    VmecExtenderEdgeFieldCampaignArtifacts,
+    build_vmec_extender_edge_field_campaign_report,
+    create_vmec_extender_edge_field_campaign_package,
+    save_vmec_extender_edge_field_campaign_plot,
+)
+from .vmec_extender_sol_smoke_campaign import (
+    VmecExtenderSolSmokeArtifacts,
+    VmecExtenderSolSmokeResult,
+    build_vmec_extender_sol_smoke_report,
+    create_vmec_extender_sol_smoke_package,
+    save_vmec_extender_sol_smoke_plot,
+    simulate_vmec_extender_scalar_sol_smoke,
+)
+from .essos_imported_fci_campaign import (
+    EssosImportedConnectionLengthLevels,
+    EssosImportedFciCampaignArtifacts,
+    EssosImportedConnectionLengthRefinementArtifacts,
+    EssosImportedEndpointLabelLevels,
+    EssosImportedEndpointLabelRefinementArtifacts,
+    build_essos_imported_connection_length_refinement_diagnostics,
+    build_essos_imported_connection_length_refinement_campaign,
+    build_essos_imported_endpoint_label_refinement_campaign,
+    build_essos_imported_endpoint_label_refinement_diagnostics,
+    build_essos_imported_fci_campaign,
+    build_essos_imported_fci_source_profile_gate,
+    build_live_essos_imported_endpoint_label_levels,
+    build_live_essos_imported_connection_length_levels,
+    create_essos_imported_connection_length_refinement_package,
+    create_essos_imported_endpoint_label_refinement_package,
+    create_essos_imported_fci_campaign_package,
+    create_live_essos_imported_connection_length_refinement_package,
+    create_live_essos_imported_endpoint_label_refinement_package,
+    save_essos_imported_endpoint_label_refinement_plot,
+    save_essos_imported_connection_length_refinement_plot,
+    save_essos_imported_fci_campaign_plot,
+    save_essos_imported_fci_source_profile_gate_plot,
+)
+from .essos_imported_artifact_audit import (
+    audit_essos_imported_artifact_report,
+    audit_essos_imported_artifact_reports,
+    audit_hybrid_open_sol_promotion_evidence,
+)
+from .essos_imported_pytree_campaign import (
+    EssosImportedPytreeCampaignArtifacts,
+    build_essos_imported_pytree_campaign,
+    create_essos_imported_pytree_campaign_package,
+    initial_essos_imported_drb_state,
+    save_essos_imported_pytree_campaign_plot,
+)
+from .essos_imported_drb_movie_campaign import (
+    ESSOS_IMPORTED_DRB_MOVIE_REFINEMENT_METRICS,
+    EssosImportedDrbMovieArtifacts,
+    EssosImportedDrbMovieRefinementArtifacts,
+    EssosImportedDrbMovieRefinementCampaignArtifacts,
+    EssosImportedDrbMovieStationarityArtifacts,
+    build_essos_imported_drb_movie_campaign,
+    build_essos_imported_drb_movie_refinement_diagnostics,
+    build_essos_imported_drb_movie_refinement_next_campaign,
+    build_essos_imported_drb_movie_refinement_summary,
+    build_essos_imported_drb_movie_stationarity_report,
+    classify_essos_imported_drb_movie_evidence,
+    create_essos_imported_drb_movie_refinement_campaign_package,
+    create_essos_imported_drb_movie_refinement_summary_package,
+    create_essos_imported_drb_movie_package,
+    create_essos_imported_drb_movie_stationarity_package,
+    save_essos_imported_drb_3d_frame,
+    save_essos_imported_drb_3d_movie,
+    save_essos_imported_drb_diagnostics_panel,
+    save_essos_imported_drb_snapshot_panel,
+)
+from .geometry_lineouts import (
+    LineoutSpec,
+    build_lineout_report,
+    save_lineout_summary_plot,
+    write_lineout_arrays_npz,
+)
+from .geometry_slices import (
+    SliceSpec,
+    build_slice_report,
+    save_slice_gif,
+    save_slice_summary_plot,
+    write_slice_arrays_npz,
+    write_slice_report_json,
+)
+from .stellarator_fci_geometry_campaign import (
+    StellaratorFciGeometryCampaignArtifacts,
+    build_stellarator_fci_geometry_report,
+    create_stellarator_fci_geometry_campaign_package,
+    save_stellarator_fci_geometry_plot,
+)
+from .stellarator_fci_operator_campaign import (
+    StellaratorFciOperatorCampaignArtifacts,
+    build_stellarator_fci_operator_campaign,
+    create_stellarator_fci_operator_campaign_package,
+    save_stellarator_fci_operator_plot,
+)
+from .stellarator_metric_mms_campaign import (
+    StellaratorMetricMmsCampaignArtifacts,
+    build_stellarator_metric_mms_campaign,
+    create_stellarator_metric_mms_campaign_package,
+    save_stellarator_metric_mms_plot,
+)
+from .stellarator_fci_suite_campaign import (
+    StellaratorFciSuiteCampaignArtifacts,
+    build_stellarator_fci_suite_campaign,
+    create_stellarator_fci_suite_campaign_package,
+    save_stellarator_fci_suite_plot,
+)
+from .stellarator_sheath_recycling_campaign import (
+    StellaratorSheathRecyclingCampaignArtifacts,
+    build_stellarator_sheath_recycling_campaign,
+    create_stellarator_sheath_recycling_campaign_package,
+    save_stellarator_sheath_recycling_plot,
+)
+from .stellarator_neutral_physics_campaign import (
+    StellaratorNeutralPhysicsCampaignArtifacts,
+    build_stellarator_neutral_physics_campaign,
+    create_stellarator_neutral_physics_campaign_package,
+    save_stellarator_neutral_physics_plot,
+)
+from .stellarator_vorticity_campaign import (
+    StellaratorVorticityCampaignArtifacts,
+    build_stellarator_vorticity_campaign,
+    create_stellarator_vorticity_campaign_package,
+    save_stellarator_vorticity_plot,
+)
+from .stellarator_drb_pytree_campaign import (
+    StellaratorDrbPytreeCampaignArtifacts,
+    build_stellarator_drb_pytree_campaign,
+    create_stellarator_drb_pytree_campaign_package,
+    initial_fci_drb_state,
+    save_stellarator_drb_pytree_plot,
+)
+from .stellarator_sol_showcase import (
+    StellaratorSolShowcaseArtifacts,
+    build_stellarator_sol_showcase_report,
+    create_stellarator_sol_showcase_package,
+    save_stellarator_sol_3d_frame,
+    save_stellarator_sol_3d_movie,
+    save_stellarator_sol_diagnostics_panel,
+    save_stellarator_sol_snapshot_panel,
+    simulate_reduced_stellarator_sol_dynamics,
+)
+from .fluid_1d_mms_convergence import (
+    Fluid1DMmsConvergenceArtifacts,
+    build_fluid_1d_mms_convergence_report,
+    create_fluid_1d_mms_convergence_package,
+    save_fluid_1d_mms_convergence_plot,
+)
+__all__ = [
+    "AlfvenWaveAnalysisResult",
+    "AlfvenWaveBenchmarkScalars",
+    "AlfvenWaveParityResult",
+    "analyze_alfven_wave_array_payload",
+    "analyze_alfven_wave_npz",
+    "compare_alfven_wave_array_payloads",
+    "compare_alfven_wave_npz",
+    "compute_alfven_wave_benchmark_scalars",
+    "save_alfven_wave_diagnostic_plot",
+    "save_alfven_wave_parity_plot",
+    "write_alfven_wave_analysis_json",
+    "write_alfven_wave_parity_json",
+    "Blob2DAnalysisResult",
+    "Blob2DParityResult",
+    "analyze_blob2d_array_payload",
+    "analyze_blob2d_npz",
+    "compare_blob2d_analysis_results",
+    "compare_blob2d_array_payloads",
+    "compare_blob2d_artifacts",
+    "compare_blob2d_npz",
+    "load_blob2d_analysis_json",
+    "save_blob2d_parity_plot",
+    "write_blob2d_analysis_json",
+    "write_blob2d_parity_json",
+    "DriftWaveAnalysisResult",
+    "DriftWaveBenchmarkScalars",
+    "DriftWaveParityResult",
+    "DriftWaveParityVariableError",
+    "analyze_drift_wave_array_payload",
+    "analyze_drift_wave_npz",
+    "compare_drift_wave_array_payloads",
+    "compare_drift_wave_npz",
+    "compute_drift_wave_benchmark_scalars",
+    "save_drift_wave_diagnostic_plot",
+    "save_drift_wave_parity_plot",
+    "write_drift_wave_analysis_json",
+    "write_drift_wave_parity_json",
+    "NeutralMixedAnalysisResult",
+    "NeutralMixedParityResult",
+    "NeutralMixedSeriesError",
+    "NeutralMixedTermBalanceCampaignArtifacts",
+    "analyze_neutral_mixed_array_payload",
+    "analyze_neutral_mixed_npz",
+    "compare_neutral_mixed_analysis_results",
+    "compare_neutral_mixed_array_payloads",
+    "compare_neutral_mixed_artifacts",
+    "compare_neutral_mixed_npz",
+    "load_neutral_mixed_analysis_json",
+    "save_neutral_mixed_diagnostic_plot",
+    "save_neutral_mixed_parity_plot",
+    "write_neutral_mixed_analysis_json",
+    "write_neutral_mixed_parity_json",
+    "build_neutral_mixed_reference_input_closure_report",
+    "build_neutral_mixed_term_balance_campaign_report",
+    "create_neutral_mixed_term_balance_campaign_package",
+    "save_neutral_mixed_term_balance_campaign_plot",
+    "write_neutral_mixed_diagnostic_input",
+    "write_neutral_mixed_reference_input_closure_json",
+    "DiffusionAutodiffSetup",
+    "StrongScalingPoint",
+    "build_diffusion_autodiff_setup",
+    "compute_strong_scaling_points",
+    "design_field",
+    "design_field_from_physical",
+    "finite_difference_gradient",
+    "objective_for_parameter_vector",
+    "objective_for_physical_parameters",
+    "optimize_inverse_design",
+    "physical_to_theta",
+    "simulate_density_history",
+    "simulate_density_history_from_physical",
+    "theta_to_physical",
+    "AutodiffDiffusionUncertaintyArtifacts",
+    "build_autodiff_diffusion_uncertainty_report",
+    "create_autodiff_diffusion_uncertainty_package",
+    "save_autodiff_diffusion_uncertainty_plot",
+    "DivertedTokamakFieldHistory",
+    "DivertedTokamakGeometry",
+    "DivertedTokamakMovieArtifacts",
+    "assemble_tokamak_rank_history",
+    "build_diverted_tokamak_analysis",
+    "create_diverted_tokamak_movie_package",
+    "create_diverted_tokamak_movie_package_from_arrays",
+    "load_diverted_tokamak_arrays_npz",
+    "load_diverted_tokamak_geometry",
+    "save_diverted_tokamak_gif",
+    "save_diverted_tokamak_poster_frame",
+    "save_diverted_tokamak_snapshot_panel",
+    "toroidal_mean_fluctuation",
+    "write_diverted_tokamak_analysis_json",
+    "write_diverted_tokamak_arrays_npz",
+    "EssosFieldLineImportArtifacts",
+    "build_essos_fieldline_import_report",
+    "create_essos_fieldline_import_package",
+    "save_essos_fieldline_import_plot",
+    "EssosVmecFieldlineSurfaceArtifacts",
+    "build_essos_vmec_fieldline_surface_campaign",
+    "create_essos_vmec_fieldline_surface_package",
+    "save_essos_vmec_fieldline_surface_plot",
+    "EssosVmecClosedFieldArtifacts",
+    "EssosVmecClosedFieldDryRunArtifacts",
+    "build_essos_vmec_closed_field_report",
+    "create_essos_vmec_closed_field_dry_run_package",
+    "create_essos_vmec_closed_field_package",
+    "save_essos_vmec_closed_field_plot",
+    "EssosVmecClosedFieldTransientArtifacts",
+    "EssosVmecClosedFieldTransientDryRunArtifacts",
+    "build_essos_vmec_closed_field_transient_campaign",
+    "create_essos_vmec_closed_field_transient_dry_run_package",
+    "create_essos_vmec_closed_field_transient_package",
+    "create_essos_vmec_closed_field_transient_package_from_geometry",
+    "save_essos_vmec_closed_field_transient_movie",
+    "save_essos_vmec_closed_field_transient_plot",
+    "EssosImportedConnectionLengthLevels",
+    "EssosImportedConnectionLengthRefinementArtifacts",
+    "EssosImportedEndpointLabelLevels",
+    "EssosImportedEndpointLabelRefinementArtifacts",
+    "EssosImportedFciCampaignArtifacts",
+    "build_essos_imported_connection_length_refinement_campaign",
+    "build_essos_imported_connection_length_refinement_diagnostics",
+    "build_essos_imported_endpoint_label_refinement_campaign",
+    "build_essos_imported_endpoint_label_refinement_diagnostics",
+    "build_essos_imported_fci_campaign",
+    "build_essos_imported_fci_source_profile_gate",
+    "build_live_essos_imported_endpoint_label_levels",
+    "build_live_essos_imported_connection_length_levels",
+    "create_essos_imported_connection_length_refinement_package",
+    "create_essos_imported_endpoint_label_refinement_package",
+    "create_essos_imported_fci_campaign_package",
+    "create_live_essos_imported_connection_length_refinement_package",
+    "create_live_essos_imported_endpoint_label_refinement_package",
+    "save_essos_imported_endpoint_label_refinement_plot",
+    "save_essos_imported_connection_length_refinement_plot",
+    "save_essos_imported_fci_campaign_plot",
+    "save_essos_imported_fci_source_profile_gate_plot",
+    "VmecExtenderEdgeFieldCampaignArtifacts",
+    "build_vmec_extender_edge_field_campaign_report",
+    "create_vmec_extender_edge_field_campaign_package",
+    "save_vmec_extender_edge_field_campaign_plot",
+    "VmecExtenderSolSmokeArtifacts",
+    "VmecExtenderSolSmokeResult",
+    "build_vmec_extender_sol_smoke_report",
+    "create_vmec_extender_sol_smoke_package",
+    "save_vmec_extender_sol_smoke_plot",
+    "simulate_vmec_extender_scalar_sol_smoke",
+    "LineoutSpec",
+    "SliceSpec",
+    "build_lineout_report",
+    "build_slice_report",
+    "save_slice_gif",
+    "save_slice_summary_plot",
+    "save_lineout_summary_plot",
+    "write_lineout_arrays_npz",
+    "write_slice_arrays_npz",
+    "write_slice_report_json",
+    "StellaratorFciGeometryCampaignArtifacts",
+    "build_stellarator_fci_geometry_report",
+    "create_stellarator_fci_geometry_campaign_package",
+    "save_stellarator_fci_geometry_plot",
+    "StellaratorFciOperatorCampaignArtifacts",
+    "build_stellarator_fci_operator_campaign",
+    "create_stellarator_fci_operator_campaign_package",
+    "save_stellarator_fci_operator_plot",
+    "StellaratorMetricMmsCampaignArtifacts",
+    "build_stellarator_metric_mms_campaign",
+    "create_stellarator_metric_mms_campaign_package",
+    "save_stellarator_metric_mms_plot",
+    "StellaratorFciSuiteCampaignArtifacts",
+    "build_stellarator_fci_suite_campaign",
+    "create_stellarator_fci_suite_campaign_package",
+    "save_stellarator_fci_suite_plot",
+    "StellaratorSheathRecyclingCampaignArtifacts",
+    "build_stellarator_sheath_recycling_campaign",
+    "create_stellarator_sheath_recycling_campaign_package",
+    "save_stellarator_sheath_recycling_plot",
+    "StellaratorNeutralPhysicsCampaignArtifacts",
+    "build_stellarator_neutral_physics_campaign",
+    "create_stellarator_neutral_physics_campaign_package",
+    "save_stellarator_neutral_physics_plot",
+    "StellaratorVorticityCampaignArtifacts",
+    "build_stellarator_vorticity_campaign",
+    "create_stellarator_vorticity_campaign_package",
+    "save_stellarator_vorticity_plot",
+    "StellaratorDrbPytreeCampaignArtifacts",
+    "build_stellarator_drb_pytree_campaign",
+    "create_stellarator_drb_pytree_campaign_package",
+    "initial_fci_drb_state",
+    "save_stellarator_drb_pytree_plot",
+    "StellaratorSolShowcaseArtifacts",
+    "build_stellarator_sol_showcase_report",
+    "create_stellarator_sol_showcase_package",
+    "save_stellarator_sol_3d_frame",
+    "save_stellarator_sol_3d_movie",
+    "save_stellarator_sol_diagnostics_panel",
+    "save_stellarator_sol_snapshot_panel",
+    "simulate_reduced_stellarator_sol_dynamics",
+    "RecyclingBatchedJvpProblem",
+    "build_recycling_batched_jvp_problem",
+    "create_recycling_batched_jvp_profile_package",
+    "profile_recycling_batched_jvp_problem",
+]
+
+# Only export what actually imported (removed-model blocks stripped in the v2 slim).
+__all__ = [name for name in __all__ if name in globals()]