omniopt2 7345__tar.gz → 7348__tar.gz

{omniopt2-7345 → omniopt2-7348}/.omniopt.py

@@ -3692,10 +3692,8 @@ def _write_job_infos_csv_result_to_strlist(result: Optional[Union[Dict[str, Opti
         for rkey in result:
             result_values.append(str(rkey))
     elif isinstance(result, dict):
-        result_keys = list(result.keys())
-        for rkey in result_keys:
-            rval = str(result[str(rkey)])
-            result_values.append(rval)
+        for _rkey, rval in result.items():  # type: str, Optional[float]
+            result_values.append(str(rval))
     elif result is not None:  # int or float
         result_values.append(str(result))
 

{omniopt2-7345 → omniopt2-7348}/.tpe.py

@@ -6,6 +6,12 @@ from typing import Optional
 try:
     import optuna
     from optuna.trial import create_trial
+
+    from optuna.distributions import (
+        BaseDistribution,
+        IntUniformDistribution,
+        FloatDistribution,        # Optuna ≥3.6
+    )
 except ModuleNotFoundError:
     print("Optuna not found. Cannot continue.")
     sys.exit(1)
@@ -121,8 +127,8 @@ def add_existing_trial_to_study(study: optuna.study.study.Study, trial_entry: li
 
     final_value = result_dict[result_key]
 
-    trial_params = {}
-    trial_distributions = {}
+    trial_params: dict[str, object] = {}
+    trial_distributions: dict[str, BaseDistribution] = {}   # 👈 explicit & correct
 
     for name, p in parameters.items():
         value = param_dict[name]
@@ -131,26 +137,27 @@ def add_existing_trial_to_study(study: optuna.study.study.Study, trial_entry: li
             trial_params[name] = value
             continue
 
+        dist: BaseDistribution
         if p["parameter_type"] == "RANGE":
             if p["type"] == "INT":
-                dist = optuna.distributions.IntUniformDistribution(p["range"][0], p["range"][1])
+                dist = IntUniformDistribution(p["range"][0], p["range"][1])
             elif p["type"] == "FLOAT":
-                dist = optuna.distributions.UniformDistribution(p["range"][0], p["range"][1])
+                # pick the right class for your Optuna version
+                dist = FloatDistribution(p["range"][0], p["range"][1])
             else:
                 continue
-
         elif p["parameter_type"] == "CHOICE":
             dist = optuna.distributions.CategoricalDistribution(p["values"])
         else:
             continue
 
         trial_params[name] = value
-        trial_distributions[name] = dist
+        trial_distributions[name] = dist      # keys are str, values are BaseDistribution
 
     study.add_trial(
         create_trial(
             params=trial_params,
-            distributions=trial_distributions,
+            distributions=trial_distributions,  # ✅ mypy is happy now
             value=final_value
         )
     )

{omniopt2-7345 → omniopt2-7348}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: omniopt2
-Version: 7345
+Version: 7348
 Summary: Automatic highly parallelized hyperparameter optimizer based on Ax/Botorch
 Home-page: https://scads.ai/transfer-2/verfuegbare-software-dienste-en/omniopt/
 Author: Norman Koch

{omniopt2-7345 → omniopt2-7348}/omniopt2.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: omniopt2
-Version: 7345
+Version: 7348
 Summary: Automatic highly parallelized hyperparameter optimizer based on Ax/Botorch
 Home-page: https://scads.ai/transfer-2/verfuegbare-software-dienste-en/omniopt/
 Author: Norman Koch

{omniopt2-7345 → omniopt2-7348}/pyproject.toml

@@ -5,7 +5,7 @@ authors = [
   {email = "norman.koch@tu-dresden.de"},
   {name = "Norman Koch"}
 ]
-version = "7345"
+version = "7348"
 
 readme = "README.md"
 dynamic = ["dependencies"]