Save TB logs as part of push_to_hub (#27022)
* Support runs/ * Upload runs folder as part of push to hub * Add a test * Add to test deps * Update with proposed solution from Slack * Ensure that repo gets deleted in tests
This commit is contained in:
@@ -45,6 +45,7 @@ from .integrations import (
|
||||
is_optuna_available,
|
||||
is_ray_available,
|
||||
is_sigopt_available,
|
||||
is_tensorboard_available,
|
||||
is_wandb_available,
|
||||
)
|
||||
from .integrations.deepspeed import is_deepspeed_available
|
||||
@@ -911,6 +912,13 @@ def require_optimum(test_case):
|
||||
return unittest.skipUnless(is_optimum_available(), "test requires optimum")(test_case)
|
||||
|
||||
|
||||
def require_tensorboard(test_case):
|
||||
"""
|
||||
Decorator for `tensorboard` dependency
|
||||
"""
|
||||
return unittest.skipUnless(is_tensorboard_available(), "test requires tensorboard")
|
||||
|
||||
|
||||
def require_auto_gptq(test_case):
|
||||
"""
|
||||
Decorator for auto_gptq dependency
|
||||
|
||||
Reference in New Issue
Block a user