omniopt2 8129__py3-none-any.whl → 8167__py3-none-any.whl

.omniopt.py

@@ -65,6 +65,7 @@ _func_call_paths = defaultdict(Counter)
 _last_mem = defaultdict(float)
 _leak_threshold_mb = 10.0
 generation_strategy_names: list = []
+default_max_range_difference: int = 1000000
 
 _function_name_cache: dict = {}
 
@@ -130,7 +131,7 @@ try:
     )
 
     def spinner(text: str) -> Any:
-        return console.status(f"[bold green]{text}")
+        return console.status(f"[bold green]{text}", speed=0.2, refresh_per_second=6)
 
     with spinner("Importing logging..."):
         import logging
@@ -662,6 +663,15 @@ def my_exit(_code: int = 0) -> None:
 
     print(f"Wallclock-Runtime: {whole_run_time} seconds {human_time}")
 
+    if is_skip_search() and os.getenv("SKIP_SEARCH_EXIT_CODE"):
+        skip_search_exit_code = os.getenv("SKIP_SEARCH_EXIT_CODE")
+
+        try:
+            sys.exit(int(skip_search_exit_code))
+        except ValueError:
+            print(f"Trying to look for SKIP_SEARCH_EXIT_CODE failed. Exiting with original exit code {_code}")
+            sys.exit(_code)
+
     sys.exit(_code)
 
 def print_green(text: str) -> None:
@@ -817,6 +827,8 @@ class ConfigLoader:
     occ: bool
     force_choice_for_ranges: bool
     dryrun: bool
+    range_max_difference: int
+    skip_search: bool
     just_return_defaults: bool
     run_mode: str
 
@@ -901,6 +913,8 @@ class ConfigLoader:
         optional.add_argument('--run_program_once', type=str, help='Path to a setup script that will run once before the main program starts.')
         optional.add_argument('--worker_generator_path', type=str, help='Path of the run folder where this script should plug itself in as a worker points generator')
         optional.add_argument('--save_to_database', help='Save all entries into a sqlite3 database', action='store_true', default=False)
+        optional.add_argument('--range_max_difference', help=f'Max. difference for range, default is {default_max_range_difference}', default=default_max_range_difference, type=int)
+        optional.add_argument('--skip_search', help='Skips the actual search, uses exit code 0 if not the environment variable SKIP_SEARCH_EXIT_CODE is set', action='store_true', default=False)
 
         speed.add_argument('--dont_warm_start_refitting', help='Do not keep Model weights, thus, refit for every generator (may be more accurate, but slower)', action='store_true', default=False)
         speed.add_argument('--refit_on_cv', help='Refit on Cross-Validation (helps in accuracy, but makes generating new points slower)', action='store_true', default=False)
@@ -966,7 +980,8 @@ class ConfigLoader:
                 if file_format == 'json':
                     return json.load(file)
             except (Exception, json.decoder.JSONDecodeError) as e:
-                print_red(f"Error parsing {file_format} file '{config_path}': {e}")
+                print_red(f"Error parsing {file_format} file '{config_path}'")
+                print_debug(f"Error parsing {file_format} file {config_path}: {e}")
                 self._parsing_arguments_loader.stop()
                 print("Exit-Code: 5")
                 sys.exit(5)
@@ -1142,6 +1157,15 @@ with spinner("Parsing arguments...") as parsing_arguments_loader:
     loader = ConfigLoader(parsing_arguments_loader)
     args = loader.parse_arguments()
 
+def is_skip_search() -> bool:
+    if args.skip_search:
+        return True
+
+    if os.getenv("SKIP_SEARCH"):
+        return True
+
+    return False
+
 original_result_names = args.result_names
 
 if args.seed is not None:
@@ -1606,7 +1630,7 @@ class InteractiveCLIGenerationNode(ExternalGenerationNode):
     constraints: Optional[Any]
     fit_time_since_gen: float
 
-    def __init__(                  # identical signature to the original class
+    def __init__(
         self: Any,
         node_name: str = "INTERACTIVE_GENERATOR",
     ) -> None:
@@ -2206,7 +2230,7 @@ class SearchSpaceExhausted (Exception):
     pass
 
 NR_INSERTED_JOBS: int = 0
-executor: Union[LocalExecutor, AutoExecutor, None] = None
+submitit_executor: Union[LocalExecutor, AutoExecutor, None] = None
 
 NR_OF_0_RESULTS: int = 0
 
@@ -3093,6 +3117,14 @@ def get_value_type_and_log_scale(this_args: Union[str, list], j: int) -> Tuple[i
 
     return skip, value_type, log_scale
 
+def check_for_too_high_differences(lower_bound: Union[int, float], upper_bound: Union[int, float]) -> None:
+    bound_diff = abs(lower_bound - upper_bound)
+
+    if bound_diff > args.range_max_difference:
+        print_red(f"The difference between {lower_bound} and {upper_bound} was too high, these large numbers can cause memory leaks. Difference was: {bound_diff}, max difference is {args.range_max_difference}")
+
+        sys.exit(235)
+
 def parse_range_param(classic_params: list, params: list, j: int, this_args: Union[str, list], name: str, search_space_reduction_warning: bool) -> Tuple[int, list, list, bool]:
     check_factorial_range()
     check_range_params_length(this_args)
@@ -3102,7 +3134,7 @@ def parse_range_param(classic_params: list, params: list, j: int, this_args: Uni
 
     lower_bound, upper_bound = get_bounds(this_args, j)
 
-    die_181_or_91_if_lower_and_upper_bound_equal_zero(lower_bound, upper_bound)
+    die_if_lower_and_upper_bound_equal_zero(lower_bound, upper_bound)
 
     lower_bound, upper_bound = switch_lower_and_upper_if_needed(name, lower_bound, upper_bound)
 
@@ -3114,6 +3146,12 @@ def parse_range_param(classic_params: list, params: list, j: int, this_args: Uni
 
     lower_bound, upper_bound = get_bounds_from_previous_data(name, lower_bound, upper_bound)
 
+    check_for_too_high_differences(lower_bound, upper_bound)
+
+    if lower_bound == upper_bound:
+        print_red(f"Lower bound {lower_bound} was equal to upper bound {upper_bound}. Please fix this. Cannot continue.")
+        my_exit(181)
+
     search_space_reduction_warning = check_bounds_change_due_to_previous_job(name, lower_bound, upper_bound, search_space_reduction_warning)
 
     param = create_range_param(name, lower_bound, upper_bound, value_type, log_scale)
@@ -3271,9 +3309,9 @@ def check_range_params_length(this_args: Union[str, list]) -> None:
     if len(this_args) != 5 and len(this_args) != 4 and len(this_args) != 6:
         _fatal_error("\n⚠ --parameter for type range must have 4 (or 5, the last one being optional and float by default, or 6, while the last one is true or false) parameters: <NAME> range <START> <END> (<TYPE (int or float)>, <log_scale: bool>)", 181)
 
-def die_181_or_91_if_lower_and_upper_bound_equal_zero(lower_bound: Union[int, float], upper_bound: Union[int, float]) -> None:
+def die_if_lower_and_upper_bound_equal_zero(lower_bound: Union[int, float], upper_bound: Union[int, float]) -> None:
     if upper_bound is None or lower_bound is None:
-        _fatal_error("die_181_or_91_if_lower_and_upper_bound_equal_zero: upper_bound or lower_bound is None. Cannot continue.", 91)
+        _fatal_error("die_if_lower_and_upper_bound_equal_zero: upper_bound or lower_bound is None. Cannot continue.", 91)
     if upper_bound == lower_bound:
         if lower_bound == 0:
             _fatal_error(f"⚠ Lower bound and upper bound are equal: {lower_bound}, cannot automatically fix this, because they -0 = +0 (usually a quickfix would be to set lower_bound = -upper_bound)", 181)
@@ -7331,7 +7369,7 @@ def submit_new_job(parameters: Union[dict, str], trial_index: int) -> Any:
 
     start = time.time()
 
-    new_job = executor.submit(evaluate, {"params": parameters, "trial_idx": trial_index, "submit_time": int(time.time())})
+    new_job = submitit_executor.submit(evaluate, {"params": parameters, "trial_idx": trial_index, "submit_time": int(time.time())})
 
     elapsed = time.time() - start
 
@@ -7340,7 +7378,7 @@ def submit_new_job(parameters: Union[dict, str], trial_index: int) -> Any:
     return new_job
 
 def orchestrator_start_trial(parameters: Union[dict, str], trial_index: int) -> None:
-    if executor and ax_client:
+    if submitit_executor and ax_client:
         new_job = submit_new_job(parameters, trial_index)
         submitted_jobs(1)
 
@@ -7355,7 +7393,7 @@ def orchestrator_start_trial(parameters: Union[dict, str], trial_index: int) ->
         print_debug(f"orchestrator_start_trial: appending job {new_job} to global_vars['jobs'], trial_index: {trial_index}")
         global_vars["jobs"].append((new_job, trial_index))
     else:
-        _fatal_error("executor or ax_client could not be found properly", 9)
+        _fatal_error("submitit_executor or ax_client could not be found properly", 9)
 
 def handle_exclude_node(stdout_path: str, hostname_from_out_file: Union[None, str]) -> None:
     stdout_path = check_alternate_path(stdout_path)
@@ -7465,8 +7503,8 @@ def execute_evaluation(_params: list) -> Optional[int]:
 
         return None
 
-    if not executor:
-        _fatal_error("executor could not be found", 9)
+    if not submitit_executor:
+        _fatal_error("submitit_executor could not be found", 9)
 
         return None
 
@@ -7521,10 +7559,10 @@ def set_sbatch_environment() -> None:
 def exclude_defective_nodes() -> None:
     excluded_string: str = ",".join(count_defective_nodes())
     if len(excluded_string) > 1:
-        if executor:
-            executor.update_parameters(exclude=excluded_string)
+        if submitit_executor:
+            submitit_executor.update_parameters(exclude=excluded_string)
         else:
-            _fatal_error("executor could not be found", 9)
+            _fatal_error("submitit_executor could not be found", 9)
 
 def handle_failed_job(error: Union[None, Exception, str], trial_index: int, new_job: Optional[Job]) -> None:
     if "QOSMinGRES" in str(error) and args.gpus == 0:
@@ -8821,6 +8859,10 @@ def _create_and_execute_next_runs_run_loop(_max_eval: Optional[int], phase: Opti
 
             filtered_trial_index_to_param = {k: v for k, v in trial_index_to_param.items() if k not in abandoned_trial_indices}
 
+            if is_skip_search():
+                print_yellow("Skipping search part")
+                return True, {}
+
             if len(filtered_trial_index_to_param):
                 execute_trials(filtered_trial_index_to_param, phase, _max_eval)
             else:
@@ -8873,17 +8915,17 @@ def get_number_of_steps(_max_eval: int) -> int:
         return _random_steps
 
 def _set_global_executor() -> None:
-    global executor
+    global submitit_executor
 
     log_folder: str = f'{get_current_run_folder()}/single_runs/%j'
     subjob_uuid = str(uuid.uuid4())
 
     if args.force_local_execution:
-        executor = LocalExecutor(folder=log_folder)
+        submitit_executor = LocalExecutor(folder=log_folder)
     else:
-        executor = AutoExecutor(folder=log_folder)
+        submitit_executor = AutoExecutor(folder=log_folder)
 
-    if executor:
+    if submitit_executor:
         params = {
             "name": f'{global_vars["experiment_name"]}_{run_uuid}_{subjob_uuid}',
             "timeout_min": args.worker_timeout,
@@ -8897,14 +8939,14 @@ def _set_global_executor() -> None:
             "exclude": args.exclude,
         }
 
-        executor.update_parameters(**params)
+        submitit_executor.update_parameters(**params)
 
-        print_debug("executor.update_parameters(\n" + json.dumps(params, indent=4) + "\n)")
+        print_debug("submitit_executor.update_parameters(\n" + json.dumps(params, indent=4) + "\n)")
 
         if args.exclude:
             print_yellow(f"Excluding the following nodes: {args.exclude}")
     else:
-        _fatal_error("executor could not be found", 9)
+        _fatal_error("submitit_executor could not be found", 9)
 
 def set_global_executor() -> None:
     try:
@@ -10263,7 +10305,13 @@ def run_program_once(params=None) -> None:
         my_exit(57)
 
 def show_omniopt_call() -> None:
-    original_print(oo_call + " " + " ".join(sys.argv[1:]))
+    def remove_ui_url(arg_str):
+        return re.sub(r'(?:--ui_url(?:=\S+)?(?:\s+\S+)?)', '', arg_str).strip()
+
+    original_argv = " ".join(sys.argv[1:])
+    cleaned = remove_ui_url(original_argv)
+
+    original_print(oo_call + " " + cleaned)
 
 def main() -> None:
     global RESULT_CSV_FILE, ax_client, LOGFILE_DEBUG_GET_NEXT_TRIALS
@@ -11006,7 +11054,7 @@ def main_outside() -> None:
                 end_program(True)
 
 def auto_wrap_namespace(namespace: Any) -> Any:
-    enable_beartype = os.getenv("ENABLE_BEARTYPE") is not None
+    enable_beartype = any(os.getenv(v) for v in ("ENABLE_BEARTYPE", "CI"))
 
     excluded_functions = {
         "log_time_and_memory_wrapper",
@@ -11026,7 +11074,6 @@ def auto_wrap_namespace(namespace: Any) -> Any:
         if (isinstance(obj, types.FunctionType) and name not in excluded_functions):
             wrapped = obj
             if enable_beartype:
-                print(f"Wrapping function '{name}' in @beartype")
                 wrapped = beartype(wrapped)
 
             if args.runtime_debug:

omniopt

@@ -1954,7 +1954,7 @@ EOF
 			sbatch_command=$(echo "$sbatch_command" | sed -e 's#^[[:space:]]*##' -e 's#[[:space:]]*$##')
 			red_text "Failed to start sbatch job. Command:\n"
 			red_text "$sbatch_command\n"
-			red_text "Exit-Code: $exit_code\n"
+			red_text "Exit-Code (sbatch): $exit_code\n"
 
 			myexit $exit_code
 		fi

{omniopt2-8129.data → omniopt2-8167.data}/data/bin/.omniopt.py

@@ -65,6 +65,7 @@ _func_call_paths = defaultdict(Counter)
 _last_mem = defaultdict(float)
 _leak_threshold_mb = 10.0
 generation_strategy_names: list = []
+default_max_range_difference: int = 1000000
 
 _function_name_cache: dict = {}
 
@@ -130,7 +131,7 @@ try:
     )
 
     def spinner(text: str) -> Any:
-        return console.status(f"[bold green]{text}")
+        return console.status(f"[bold green]{text}", speed=0.2, refresh_per_second=6)
 
     with spinner("Importing logging..."):
         import logging
@@ -662,6 +663,15 @@ def my_exit(_code: int = 0) -> None:
 
     print(f"Wallclock-Runtime: {whole_run_time} seconds {human_time}")
 
+    if is_skip_search() and os.getenv("SKIP_SEARCH_EXIT_CODE"):
+        skip_search_exit_code = os.getenv("SKIP_SEARCH_EXIT_CODE")
+
+        try:
+            sys.exit(int(skip_search_exit_code))
+        except ValueError:
+            print(f"Trying to look for SKIP_SEARCH_EXIT_CODE failed. Exiting with original exit code {_code}")
+            sys.exit(_code)
+
     sys.exit(_code)
 
 def print_green(text: str) -> None:
@@ -817,6 +827,8 @@ class ConfigLoader:
     occ: bool
     force_choice_for_ranges: bool
     dryrun: bool
+    range_max_difference: int
+    skip_search: bool
     just_return_defaults: bool
     run_mode: str
 
@@ -901,6 +913,8 @@ class ConfigLoader:
         optional.add_argument('--run_program_once', type=str, help='Path to a setup script that will run once before the main program starts.')
         optional.add_argument('--worker_generator_path', type=str, help='Path of the run folder where this script should plug itself in as a worker points generator')
         optional.add_argument('--save_to_database', help='Save all entries into a sqlite3 database', action='store_true', default=False)
+        optional.add_argument('--range_max_difference', help=f'Max. difference for range, default is {default_max_range_difference}', default=default_max_range_difference, type=int)
+        optional.add_argument('--skip_search', help='Skips the actual search, uses exit code 0 if not the environment variable SKIP_SEARCH_EXIT_CODE is set', action='store_true', default=False)
 
         speed.add_argument('--dont_warm_start_refitting', help='Do not keep Model weights, thus, refit for every generator (may be more accurate, but slower)', action='store_true', default=False)
         speed.add_argument('--refit_on_cv', help='Refit on Cross-Validation (helps in accuracy, but makes generating new points slower)', action='store_true', default=False)
@@ -966,7 +980,8 @@ class ConfigLoader:
                 if file_format == 'json':
                     return json.load(file)
             except (Exception, json.decoder.JSONDecodeError) as e:
-                print_red(f"Error parsing {file_format} file '{config_path}': {e}")
+                print_red(f"Error parsing {file_format} file '{config_path}'")
+                print_debug(f"Error parsing {file_format} file {config_path}: {e}")
                 self._parsing_arguments_loader.stop()
                 print("Exit-Code: 5")
                 sys.exit(5)
@@ -1142,6 +1157,15 @@ with spinner("Parsing arguments...") as parsing_arguments_loader:
     loader = ConfigLoader(parsing_arguments_loader)
     args = loader.parse_arguments()
 
+def is_skip_search() -> bool:
+    if args.skip_search:
+        return True
+
+    if os.getenv("SKIP_SEARCH"):
+        return True
+
+    return False
+
 original_result_names = args.result_names
 
 if args.seed is not None:
@@ -1606,7 +1630,7 @@ class InteractiveCLIGenerationNode(ExternalGenerationNode):
     constraints: Optional[Any]
     fit_time_since_gen: float
 
-    def __init__(                  # identical signature to the original class
+    def __init__(
         self: Any,
         node_name: str = "INTERACTIVE_GENERATOR",
     ) -> None:
@@ -2206,7 +2230,7 @@ class SearchSpaceExhausted (Exception):
     pass
 
 NR_INSERTED_JOBS: int = 0
-executor: Union[LocalExecutor, AutoExecutor, None] = None
+submitit_executor: Union[LocalExecutor, AutoExecutor, None] = None
 
 NR_OF_0_RESULTS: int = 0
 
@@ -3093,6 +3117,14 @@ def get_value_type_and_log_scale(this_args: Union[str, list], j: int) -> Tuple[i
 
     return skip, value_type, log_scale
 
+def check_for_too_high_differences(lower_bound: Union[int, float], upper_bound: Union[int, float]) -> None:
+    bound_diff = abs(lower_bound - upper_bound)
+
+    if bound_diff > args.range_max_difference:
+        print_red(f"The difference between {lower_bound} and {upper_bound} was too high, these large numbers can cause memory leaks. Difference was: {bound_diff}, max difference is {args.range_max_difference}")
+
+        sys.exit(235)
+
 def parse_range_param(classic_params: list, params: list, j: int, this_args: Union[str, list], name: str, search_space_reduction_warning: bool) -> Tuple[int, list, list, bool]:
     check_factorial_range()
     check_range_params_length(this_args)
@@ -3102,7 +3134,7 @@ def parse_range_param(classic_params: list, params: list, j: int, this_args: Uni
 
     lower_bound, upper_bound = get_bounds(this_args, j)
 
-    die_181_or_91_if_lower_and_upper_bound_equal_zero(lower_bound, upper_bound)
+    die_if_lower_and_upper_bound_equal_zero(lower_bound, upper_bound)
 
     lower_bound, upper_bound = switch_lower_and_upper_if_needed(name, lower_bound, upper_bound)
 
@@ -3114,6 +3146,12 @@ def parse_range_param(classic_params: list, params: list, j: int, this_args: Uni
 
     lower_bound, upper_bound = get_bounds_from_previous_data(name, lower_bound, upper_bound)
 
+    check_for_too_high_differences(lower_bound, upper_bound)
+
+    if lower_bound == upper_bound:
+        print_red(f"Lower bound {lower_bound} was equal to upper bound {upper_bound}. Please fix this. Cannot continue.")
+        my_exit(181)
+
     search_space_reduction_warning = check_bounds_change_due_to_previous_job(name, lower_bound, upper_bound, search_space_reduction_warning)
 
     param = create_range_param(name, lower_bound, upper_bound, value_type, log_scale)
@@ -3271,9 +3309,9 @@ def check_range_params_length(this_args: Union[str, list]) -> None:
     if len(this_args) != 5 and len(this_args) != 4 and len(this_args) != 6:
         _fatal_error("\n⚠ --parameter for type range must have 4 (or 5, the last one being optional and float by default, or 6, while the last one is true or false) parameters: <NAME> range <START> <END> (<TYPE (int or float)>, <log_scale: bool>)", 181)
 
-def die_181_or_91_if_lower_and_upper_bound_equal_zero(lower_bound: Union[int, float], upper_bound: Union[int, float]) -> None:
+def die_if_lower_and_upper_bound_equal_zero(lower_bound: Union[int, float], upper_bound: Union[int, float]) -> None:
     if upper_bound is None or lower_bound is None:
-        _fatal_error("die_181_or_91_if_lower_and_upper_bound_equal_zero: upper_bound or lower_bound is None. Cannot continue.", 91)
+        _fatal_error("die_if_lower_and_upper_bound_equal_zero: upper_bound or lower_bound is None. Cannot continue.", 91)
     if upper_bound == lower_bound:
         if lower_bound == 0:
             _fatal_error(f"⚠ Lower bound and upper bound are equal: {lower_bound}, cannot automatically fix this, because they -0 = +0 (usually a quickfix would be to set lower_bound = -upper_bound)", 181)
@@ -7331,7 +7369,7 @@ def submit_new_job(parameters: Union[dict, str], trial_index: int) -> Any:
 
     start = time.time()
 
-    new_job = executor.submit(evaluate, {"params": parameters, "trial_idx": trial_index, "submit_time": int(time.time())})
+    new_job = submitit_executor.submit(evaluate, {"params": parameters, "trial_idx": trial_index, "submit_time": int(time.time())})
 
     elapsed = time.time() - start
 
@@ -7340,7 +7378,7 @@ def submit_new_job(parameters: Union[dict, str], trial_index: int) -> Any:
     return new_job
 
 def orchestrator_start_trial(parameters: Union[dict, str], trial_index: int) -> None:
-    if executor and ax_client:
+    if submitit_executor and ax_client:
         new_job = submit_new_job(parameters, trial_index)
         submitted_jobs(1)
 
@@ -7355,7 +7393,7 @@ def orchestrator_start_trial(parameters: Union[dict, str], trial_index: int) ->
         print_debug(f"orchestrator_start_trial: appending job {new_job} to global_vars['jobs'], trial_index: {trial_index}")
         global_vars["jobs"].append((new_job, trial_index))
     else:
-        _fatal_error("executor or ax_client could not be found properly", 9)
+        _fatal_error("submitit_executor or ax_client could not be found properly", 9)
 
 def handle_exclude_node(stdout_path: str, hostname_from_out_file: Union[None, str]) -> None:
     stdout_path = check_alternate_path(stdout_path)
@@ -7465,8 +7503,8 @@ def execute_evaluation(_params: list) -> Optional[int]:
 
         return None
 
-    if not executor:
-        _fatal_error("executor could not be found", 9)
+    if not submitit_executor:
+        _fatal_error("submitit_executor could not be found", 9)
 
         return None
 
@@ -7521,10 +7559,10 @@ def set_sbatch_environment() -> None:
 def exclude_defective_nodes() -> None:
     excluded_string: str = ",".join(count_defective_nodes())
     if len(excluded_string) > 1:
-        if executor:
-            executor.update_parameters(exclude=excluded_string)
+        if submitit_executor:
+            submitit_executor.update_parameters(exclude=excluded_string)
         else:
-            _fatal_error("executor could not be found", 9)
+            _fatal_error("submitit_executor could not be found", 9)
 
 def handle_failed_job(error: Union[None, Exception, str], trial_index: int, new_job: Optional[Job]) -> None:
     if "QOSMinGRES" in str(error) and args.gpus == 0:
@@ -8821,6 +8859,10 @@ def _create_and_execute_next_runs_run_loop(_max_eval: Optional[int], phase: Opti
 
             filtered_trial_index_to_param = {k: v for k, v in trial_index_to_param.items() if k not in abandoned_trial_indices}
 
+            if is_skip_search():
+                print_yellow("Skipping search part")
+                return True, {}
+
             if len(filtered_trial_index_to_param):
                 execute_trials(filtered_trial_index_to_param, phase, _max_eval)
             else:
@@ -8873,17 +8915,17 @@ def get_number_of_steps(_max_eval: int) -> int:
         return _random_steps
 
 def _set_global_executor() -> None:
-    global executor
+    global submitit_executor
 
     log_folder: str = f'{get_current_run_folder()}/single_runs/%j'
     subjob_uuid = str(uuid.uuid4())
 
     if args.force_local_execution:
-        executor = LocalExecutor(folder=log_folder)
+        submitit_executor = LocalExecutor(folder=log_folder)
     else:
-        executor = AutoExecutor(folder=log_folder)
+        submitit_executor = AutoExecutor(folder=log_folder)
 
-    if executor:
+    if submitit_executor:
         params = {
             "name": f'{global_vars["experiment_name"]}_{run_uuid}_{subjob_uuid}',
             "timeout_min": args.worker_timeout,
@@ -8897,14 +8939,14 @@ def _set_global_executor() -> None:
             "exclude": args.exclude,
         }
 
-        executor.update_parameters(**params)
+        submitit_executor.update_parameters(**params)
 
-        print_debug("executor.update_parameters(\n" + json.dumps(params, indent=4) + "\n)")
+        print_debug("submitit_executor.update_parameters(\n" + json.dumps(params, indent=4) + "\n)")
 
         if args.exclude:
             print_yellow(f"Excluding the following nodes: {args.exclude}")
     else:
-        _fatal_error("executor could not be found", 9)
+        _fatal_error("submitit_executor could not be found", 9)
 
 def set_global_executor() -> None:
     try:
@@ -10263,7 +10305,13 @@ def run_program_once(params=None) -> None:
         my_exit(57)
 
 def show_omniopt_call() -> None:
-    original_print(oo_call + " " + " ".join(sys.argv[1:]))
+    def remove_ui_url(arg_str):
+        return re.sub(r'(?:--ui_url(?:=\S+)?(?:\s+\S+)?)', '', arg_str).strip()
+
+    original_argv = " ".join(sys.argv[1:])
+    cleaned = remove_ui_url(original_argv)
+
+    original_print(oo_call + " " + cleaned)
 
 def main() -> None:
     global RESULT_CSV_FILE, ax_client, LOGFILE_DEBUG_GET_NEXT_TRIALS
@@ -11006,7 +11054,7 @@ def main_outside() -> None:
                 end_program(True)
 
 def auto_wrap_namespace(namespace: Any) -> Any:
-    enable_beartype = os.getenv("ENABLE_BEARTYPE") is not None
+    enable_beartype = any(os.getenv(v) for v in ("ENABLE_BEARTYPE", "CI"))
 
     excluded_functions = {
         "log_time_and_memory_wrapper",
@@ -11026,7 +11074,6 @@ def auto_wrap_namespace(namespace: Any) -> Any:
         if (isinstance(obj, types.FunctionType) and name not in excluded_functions):
             wrapped = obj
             if enable_beartype:
-                print(f"Wrapping function '{name}' in @beartype")
                 wrapped = beartype(wrapped)
 
             if args.runtime_debug:

{omniopt2-8129.data → omniopt2-8167.data}/data/bin/omniopt

@@ -1954,7 +1954,7 @@ EOF
 			sbatch_command=$(echo "$sbatch_command" | sed -e 's#^[[:space:]]*##' -e 's#[[:space:]]*$##')
 			red_text "Failed to start sbatch job. Command:\n"
 			red_text "$sbatch_command\n"
-			red_text "Exit-Code: $exit_code\n"
+			red_text "Exit-Code (sbatch): $exit_code\n"
 
 			myexit $exit_code
 		fi

{omniopt2-8129.dist-info → omniopt2-8167.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: omniopt2
-Version: 8129
+Version: 8167
 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-8129.dist-info → omniopt2-8167.dist-info}/RECORD

@@ -3,7 +3,7 @@
 .general.sh,sha256=uyGMN8xNToQ0v50KoiYxm6jRmgf0orroOaodM_Nuq30,2107
 .gitignore,sha256=GmSD6dE4AqN4_fhQUdAzydst60s3w3ePYxetTr3Et7c,3776
 .helpers.py,sha256=posn69_5fnlCqU-jRTFaED0LoeC2uuQe6r9NuqvfHm0,31456
-.omniopt.py,sha256=JRBDV3C7D4L8DMq-CPwddm_qWqO04-vaJy2LEBmTEtE,431304
+.omniopt.py,sha256=Uf7m37w65oUv6Iuhu_NWDBDJUpj7IzS7YihaybSKzCw,433380
 .omniopt_plot_cpu_ram_usage.py,sha256=DbOAmdrbcZtsMnHJgHfeRngjtv6zX5J0axyua_dYezc,3932
 .omniopt_plot_general.py,sha256=3iy-bPef8I5rTB3KRz-TuleMdgKDmVZ6c8LuNQhNwu0,6810
 .omniopt_plot_gpu_usage.py,sha256=ojxVicwSoiyl7f3c-6lLuT2EpyPcSJKEcbp75LgDY2k,5107
@@ -21,51 +21,51 @@ LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
 MANIFEST.in,sha256=L3QBXyL1TtyWxbAPsfjZdCfpeBrRuI5MSgdnvwdQ22Q,24
 README.md,sha256=WJxay4TRM8wfK_k5C01SlizLUHgKqlMwa_PQwAxj-As,3221
 apt-dependencies.txt,sha256=X5tBB8ZLW9XaFtRh8B7C2pIkSoxNNawqioDr0QZAtuM,149
-omniopt,sha256=4cZwHmYuwhwzb7gSvGvN_lEp61mPZuPeuJ-kqiGojSc,50843
+omniopt,sha256=1JkaAUOyNb1e81U47sXr2VU4smjOG0i0iWTzkeH3j5E,50852
 omniopt_docker,sha256=LWVUeyvmA5AKqAHiH9jBUkR5uZ6AHMnSy0eET7mK6E4,3602
 omniopt_evaluate,sha256=9oBh0_ikCuIz_aJQZrN0j39NDiIDYoSvEFmSVIoFjJE,23842
 omniopt_plot,sha256=ze6Dz2k0w5jyUnfatAA7dk59nzQjKC_G36tDvJGWsfA,13696
 omniopt_share,sha256=2KZcTjN_7vcxNjb4sLLiWgUc38pl2-6-Qwmgq6_nTb0,13856
-pyproject.toml,sha256=3bu_D9e_qNd0WJTgBBvmxG7YehQxEkHcU3m_-pZjh5w,397
+pyproject.toml,sha256=2rlAz7Qx9S-23CAVKKkUYxClyeyx68nLO1r8fjkCBss,397
 requirements.txt,sha256=-dJa3VXI74qUD39-aF6whZV6RTN7Dt7KE3o43PnFHqA,299
 setup.cfg,sha256=HEc8uu6NpfxG5_AVh5SvXOpEFMNKPPPxgMIAH144vT4,38
 test_requirements.txt,sha256=jpyZzAwbWR_qnoRqWvpBB5MUjIX9jVwynX2D-B-r8aA,487
 .tests/pylint.rc,sha256=ZGtZjVgJtK_6tdwgscJneS-LUeDASjSVVz6veCkp9rw,22631
-omniopt2-8129.data/data/bin/.colorfunctions.sh,sha256=xxc08V3Fh_0Je20fkJMRO14u9VCSvMyOiMaDfioEyCY,1098
-omniopt2-8129.data/data/bin/.general.sh,sha256=uyGMN8xNToQ0v50KoiYxm6jRmgf0orroOaodM_Nuq30,2107
-omniopt2-8129.data/data/bin/.helpers.py,sha256=posn69_5fnlCqU-jRTFaED0LoeC2uuQe6r9NuqvfHm0,31456
-omniopt2-8129.data/data/bin/.omniopt.py,sha256=JRBDV3C7D4L8DMq-CPwddm_qWqO04-vaJy2LEBmTEtE,431304
-omniopt2-8129.data/data/bin/.omniopt_plot_cpu_ram_usage.py,sha256=DbOAmdrbcZtsMnHJgHfeRngjtv6zX5J0axyua_dYezc,3932
-omniopt2-8129.data/data/bin/.omniopt_plot_general.py,sha256=3iy-bPef8I5rTB3KRz-TuleMdgKDmVZ6c8LuNQhNwu0,6810
-omniopt2-8129.data/data/bin/.omniopt_plot_gpu_usage.py,sha256=ojxVicwSoiyl7f3c-6lLuT2EpyPcSJKEcbp75LgDY2k,5107
-omniopt2-8129.data/data/bin/.omniopt_plot_kde.py,sha256=uRLWr72TDKvj3AqJ0O0AvkKZ1ok1O1QpXnbfQQdo0nA,6873
-omniopt2-8129.data/data/bin/.omniopt_plot_scatter.py,sha256=b0_CIqgyi6PztaUVJRL9X9XBTaOonh-yDH2hRxMGkH0,8403
-omniopt2-8129.data/data/bin/.omniopt_plot_scatter_generation_method.py,sha256=rgKY_w1E516c9UucVaEvaKd8tCnoUq9xg-RrYSDzYEQ,4289
-omniopt2-8129.data/data/bin/.omniopt_plot_scatter_hex.py,sha256=UKjw40c1eumgEcf0xqB-_SakX5PB6HD3u4VwBxbsgQo,10279
-omniopt2-8129.data/data/bin/.omniopt_plot_time_and_exit_code.py,sha256=WUyl2uI59wsC1eSX_5uJHOrqcF-s5cUDIEu8u3IFMLU,6462
-omniopt2-8129.data/data/bin/.omniopt_plot_trial_index_result.py,sha256=9ApSv3-eZXohafsCAY5jr2B0rG4XpaTloCb0TuevYSg,4577
-omniopt2-8129.data/data/bin/.omniopt_plot_worker.py,sha256=VuluQq4W6KRR5RU08dxmDSFk5mbfDRkRJQFwwcLgAGw,4524
-omniopt2-8129.data/data/bin/.random_generator.py,sha256=ezBBUXpez_QaGdpCglMcJ0KZPdQP0XdX5gnLzO1xhwU,2987
-omniopt2-8129.data/data/bin/.shellscript_functions,sha256=9Yo5j3BBLDNVPwe2uNjHbVCbj3H1v1gC5Ju8KvPXfC4,13624
-omniopt2-8129.data/data/bin/.tpe.py,sha256=xxQuTZFQHHobvZuqG8cP2y2ev8ifjlLvex3TOaNWq8w,6754
-omniopt2-8129.data/data/bin/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-omniopt2-8129.data/data/bin/apt-dependencies.txt,sha256=X5tBB8ZLW9XaFtRh8B7C2pIkSoxNNawqioDr0QZAtuM,149
-omniopt2-8129.data/data/bin/omniopt,sha256=4cZwHmYuwhwzb7gSvGvN_lEp61mPZuPeuJ-kqiGojSc,50843
-omniopt2-8129.data/data/bin/omniopt_docker,sha256=LWVUeyvmA5AKqAHiH9jBUkR5uZ6AHMnSy0eET7mK6E4,3602
-omniopt2-8129.data/data/bin/omniopt_evaluate,sha256=9oBh0_ikCuIz_aJQZrN0j39NDiIDYoSvEFmSVIoFjJE,23842
-omniopt2-8129.data/data/bin/omniopt_plot,sha256=ze6Dz2k0w5jyUnfatAA7dk59nzQjKC_G36tDvJGWsfA,13696
-omniopt2-8129.data/data/bin/omniopt_share,sha256=2KZcTjN_7vcxNjb4sLLiWgUc38pl2-6-Qwmgq6_nTb0,13856
-omniopt2-8129.data/data/bin/pylint.rc,sha256=ZGtZjVgJtK_6tdwgscJneS-LUeDASjSVVz6veCkp9rw,22631
-omniopt2-8129.data/data/bin/requirements.txt,sha256=-dJa3VXI74qUD39-aF6whZV6RTN7Dt7KE3o43PnFHqA,299
-omniopt2-8129.data/data/bin/setup.py,sha256=zg7k4jeHXEdaGWGeB1mPC14fFj-hqz1vNimnU0JcMyk,4632
-omniopt2-8129.data/data/bin/test_requirements.txt,sha256=jpyZzAwbWR_qnoRqWvpBB5MUjIX9jVwynX2D-B-r8aA,487
-omniopt2-8129.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-omniopt2.egg-info/PKG-INFO,sha256=mSFHa-No7FW746wHUCP47oMscp5iwMl6Xz_USejedjY,5740
+omniopt2-8167.data/data/bin/.colorfunctions.sh,sha256=xxc08V3Fh_0Je20fkJMRO14u9VCSvMyOiMaDfioEyCY,1098
+omniopt2-8167.data/data/bin/.general.sh,sha256=uyGMN8xNToQ0v50KoiYxm6jRmgf0orroOaodM_Nuq30,2107
+omniopt2-8167.data/data/bin/.helpers.py,sha256=posn69_5fnlCqU-jRTFaED0LoeC2uuQe6r9NuqvfHm0,31456
+omniopt2-8167.data/data/bin/.omniopt.py,sha256=Uf7m37w65oUv6Iuhu_NWDBDJUpj7IzS7YihaybSKzCw,433380
+omniopt2-8167.data/data/bin/.omniopt_plot_cpu_ram_usage.py,sha256=DbOAmdrbcZtsMnHJgHfeRngjtv6zX5J0axyua_dYezc,3932
+omniopt2-8167.data/data/bin/.omniopt_plot_general.py,sha256=3iy-bPef8I5rTB3KRz-TuleMdgKDmVZ6c8LuNQhNwu0,6810
+omniopt2-8167.data/data/bin/.omniopt_plot_gpu_usage.py,sha256=ojxVicwSoiyl7f3c-6lLuT2EpyPcSJKEcbp75LgDY2k,5107
+omniopt2-8167.data/data/bin/.omniopt_plot_kde.py,sha256=uRLWr72TDKvj3AqJ0O0AvkKZ1ok1O1QpXnbfQQdo0nA,6873
+omniopt2-8167.data/data/bin/.omniopt_plot_scatter.py,sha256=b0_CIqgyi6PztaUVJRL9X9XBTaOonh-yDH2hRxMGkH0,8403
+omniopt2-8167.data/data/bin/.omniopt_plot_scatter_generation_method.py,sha256=rgKY_w1E516c9UucVaEvaKd8tCnoUq9xg-RrYSDzYEQ,4289
+omniopt2-8167.data/data/bin/.omniopt_plot_scatter_hex.py,sha256=UKjw40c1eumgEcf0xqB-_SakX5PB6HD3u4VwBxbsgQo,10279
+omniopt2-8167.data/data/bin/.omniopt_plot_time_and_exit_code.py,sha256=WUyl2uI59wsC1eSX_5uJHOrqcF-s5cUDIEu8u3IFMLU,6462
+omniopt2-8167.data/data/bin/.omniopt_plot_trial_index_result.py,sha256=9ApSv3-eZXohafsCAY5jr2B0rG4XpaTloCb0TuevYSg,4577
+omniopt2-8167.data/data/bin/.omniopt_plot_worker.py,sha256=VuluQq4W6KRR5RU08dxmDSFk5mbfDRkRJQFwwcLgAGw,4524
+omniopt2-8167.data/data/bin/.random_generator.py,sha256=ezBBUXpez_QaGdpCglMcJ0KZPdQP0XdX5gnLzO1xhwU,2987
+omniopt2-8167.data/data/bin/.shellscript_functions,sha256=9Yo5j3BBLDNVPwe2uNjHbVCbj3H1v1gC5Ju8KvPXfC4,13624
+omniopt2-8167.data/data/bin/.tpe.py,sha256=xxQuTZFQHHobvZuqG8cP2y2ev8ifjlLvex3TOaNWq8w,6754
+omniopt2-8167.data/data/bin/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+omniopt2-8167.data/data/bin/apt-dependencies.txt,sha256=X5tBB8ZLW9XaFtRh8B7C2pIkSoxNNawqioDr0QZAtuM,149
+omniopt2-8167.data/data/bin/omniopt,sha256=1JkaAUOyNb1e81U47sXr2VU4smjOG0i0iWTzkeH3j5E,50852
+omniopt2-8167.data/data/bin/omniopt_docker,sha256=LWVUeyvmA5AKqAHiH9jBUkR5uZ6AHMnSy0eET7mK6E4,3602
+omniopt2-8167.data/data/bin/omniopt_evaluate,sha256=9oBh0_ikCuIz_aJQZrN0j39NDiIDYoSvEFmSVIoFjJE,23842
+omniopt2-8167.data/data/bin/omniopt_plot,sha256=ze6Dz2k0w5jyUnfatAA7dk59nzQjKC_G36tDvJGWsfA,13696
+omniopt2-8167.data/data/bin/omniopt_share,sha256=2KZcTjN_7vcxNjb4sLLiWgUc38pl2-6-Qwmgq6_nTb0,13856
+omniopt2-8167.data/data/bin/pylint.rc,sha256=ZGtZjVgJtK_6tdwgscJneS-LUeDASjSVVz6veCkp9rw,22631
+omniopt2-8167.data/data/bin/requirements.txt,sha256=-dJa3VXI74qUD39-aF6whZV6RTN7Dt7KE3o43PnFHqA,299
+omniopt2-8167.data/data/bin/setup.py,sha256=zg7k4jeHXEdaGWGeB1mPC14fFj-hqz1vNimnU0JcMyk,4632
+omniopt2-8167.data/data/bin/test_requirements.txt,sha256=jpyZzAwbWR_qnoRqWvpBB5MUjIX9jVwynX2D-B-r8aA,487
+omniopt2-8167.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+omniopt2.egg-info/PKG-INFO,sha256=EtVKfxYyVokfB8XD-lxn2cIq-nAoF7BxHjcX-ILV3c0,5740
 omniopt2.egg-info/SOURCES.txt,sha256=2mSVIhM8Ncyl5_s8ZFBlb6qjRDbTCQmV9FwVVjp5ROE,795
 omniopt2.egg-info/dependency_links.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 omniopt2.egg-info/requires.txt,sha256=aCvaeeNulrw3AspOfG7FYwkso3fOUXeUqioyYvEnyFI,786
 omniopt2.egg-info/top_level.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-omniopt2-8129.dist-info/METADATA,sha256=mSFHa-No7FW746wHUCP47oMscp5iwMl6Xz_USejedjY,5740
-omniopt2-8129.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-omniopt2-8129.dist-info/top_level.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-omniopt2-8129.dist-info/RECORD,,
+omniopt2-8167.dist-info/METADATA,sha256=EtVKfxYyVokfB8XD-lxn2cIq-nAoF7BxHjcX-ILV3c0,5740
+omniopt2-8167.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+omniopt2-8167.dist-info/top_level.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+omniopt2-8167.dist-info/RECORD,,

omniopt2.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: omniopt2
-Version: 8129
+Version: 8167
 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

pyproject.toml

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