From 0ca2a140a0897c152d2a8e5e84145416b65dcca8 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 1 Oct 2025 01:20:30 -0700 Subject: [PATCH] execute_with_time_limit --- unsloth/__init__.py | 1 + unsloth/models/_utils.py | 1 + 2 files changed, 2 insertions(+) diff --git a/unsloth/__init__.py b/unsloth/__init__.py index bbe728045c..068c9651df 100644 --- a/unsloth/__init__.py +++ b/unsloth/__init__.py @@ -267,6 +267,7 @@ from .trainer import * from unsloth_zoo.rl_environments import ( check_python_modules, create_locked_down_function, + execute_with_time_limit, Benchmarker, ) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 328049af20..902d3ee6a8 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -94,6 +94,7 @@ from unsloth_zoo.tokenizer_utils import ( from unsloth_zoo.rl_environments import ( check_python_modules, create_locked_down_function, + execute_with_time_limit, Benchmarker, ) from unsloth_zoo.patching_utils import (