Skip to content

Commit f3f9ea9

Browse files
Further Changes to Follow Pyright
1 parent e20f852 commit f3f9ea9

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

nerfstudio/utils/writer.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -207,7 +207,7 @@ def setup_event_writer(
207207
log_dir: Path,
208208
experiment_name: str,
209209
project_name: str = "nerfstudio-project",
210-
workspace_name: str = None,
210+
workspace_name: Optional[str] = None,
211211
) -> None:
212212
"""Initialization of all event writers specified in config
213213
Args:
@@ -311,7 +311,11 @@ class WandbWriter(Writer):
311311
"""WandDB Writer Class"""
312312

313313
def __init__(
314-
self, log_dir: Path, experiment_name: str, project_name: str = "nerfstudio-project", workspace_name: str = None
314+
self,
315+
log_dir: Path,
316+
experiment_name: str,
317+
project_name: str = "nerfstudio-project",
318+
workspace_name: Optional[str] = None
315319
):
316320
import wandb # wandb is slow to import, so we only import it if we need it.
317321

@@ -377,7 +381,11 @@ class CometWriter(Writer):
377381
"""Comet_ML Writer Class"""
378382

379383
def __init__(
380-
self, log_dir: Path, experiment_name: str, project_name: str = "nerfstudio-project", workspace_name: str = None
384+
self,
385+
log_dir: Path,
386+
experiment_name: str,
387+
project_name: str = "nerfstudio-project",
388+
workspace_name: Optional[str] = None
381389
):
382390
# comet_ml is slow to import, so we only do it if we need it.
383391
import comet_ml

0 commit comments

Comments
 (0)