Skip to content

Commit e20f852

Browse files
Changes to follow Ruff Formatter
1 parent a2a9b69 commit e20f852

File tree

1 file changed

+12
-4
lines changed

1 file changed

+12
-4
lines changed

nerfstudio/utils/writer.py

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -218,11 +218,15 @@ def setup_event_writer(
218218
using_event_writer = False
219219

220220
if is_comet_enabled:
221-
curr_writer = CometWriter(log_dir=log_dir, experiment_name=experiment_name, project_name=project_name, workspace_name=workspace_name)
221+
curr_writer = CometWriter(
222+
log_dir=log_dir, experiment_name=experiment_name, project_name=project_name, workspace_name=workspace_name
223+
)
222224
EVENT_WRITERS.append(curr_writer)
223225
using_event_writer = True
224226
if is_wandb_enabled:
225-
curr_writer = WandbWriter(log_dir=log_dir, experiment_name=experiment_name, project_name=project_name, workspace_name=workspace_name)
227+
curr_writer = WandbWriter(
228+
log_dir=log_dir, experiment_name=experiment_name, project_name=project_name, workspace_name=workspace_name
229+
)
226230
EVENT_WRITERS.append(curr_writer)
227231
using_event_writer = True
228232
if is_tensorboard_enabled:
@@ -306,7 +310,9 @@ def __exit__(self, *args):
306310
class WandbWriter(Writer):
307311
"""WandDB Writer Class"""
308312

309-
def __init__(self, log_dir: Path, experiment_name: str, project_name: str = "nerfstudio-project", workspace_name: str = None):
313+
def __init__(
314+
self, log_dir: Path, experiment_name: str, project_name: str = "nerfstudio-project", workspace_name: str = None
315+
):
310316
import wandb # wandb is slow to import, so we only import it if we need it.
311317

312318
opt_args = {
@@ -370,7 +376,9 @@ def write_config(self, name: str, config_dict: Dict[str, Any], step: int):
370376
class CometWriter(Writer):
371377
"""Comet_ML Writer Class"""
372378

373-
def __init__(self, log_dir: Path, experiment_name: str, project_name: str = "nerfstudio-project", workspace_name: str = None):
379+
def __init__(
380+
self, log_dir: Path, experiment_name: str, project_name: str = "nerfstudio-project", workspace_name: str = None
381+
):
374382
# comet_ml is slow to import, so we only do it if we need it.
375383
import comet_ml
376384

0 commit comments

Comments
 (0)