mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-22 12:09:42 +00:00
remove require_nightly_gpu
This commit is contained in:
@@ -167,22 +167,6 @@ def require_package_arg(func):
|
|||||||
return wrapper
|
return wrapper
|
||||||
|
|
||||||
|
|
||||||
def require_nightly_gpu(func):
|
|
||||||
"""
|
|
||||||
Decorator that skips the test if GPU is not available.
|
|
||||||
Renamed from require_nightly_gpu to maintain backward compatibility,
|
|
||||||
but now only requires GPU availability (not nightly workflow).
|
|
||||||
"""
|
|
||||||
|
|
||||||
@require_cuda
|
|
||||||
@wraps(func)
|
|
||||||
def wrapper(*args, **kwargs):
|
|
||||||
# Only check for GPU availability, no longer require nightly workflow
|
|
||||||
return func(*args, **kwargs)
|
|
||||||
|
|
||||||
return wrapper
|
|
||||||
|
|
||||||
|
|
||||||
def require_package(package_name):
|
def require_package(package_name):
|
||||||
"""
|
"""
|
||||||
Decorator that skips the test if the specified package is not installed.
|
Decorator that skips the test if the specified package is not installed.
|
||||||
|
|||||||
Reference in New Issue
Block a user