MCPcopy
hub / github.com/SwanHubX/SwanLab / __init__

Method __init__

swanlab/integration/sb3.py:54–99  ·  view source on GitHub ↗
(
        self,
        *,
        project: Optional[str] = None,
        workspace: Optional[str] = None,
        experiment_name: Optional[str] = None,
        description: Optional[str] = None,
        log_dir: Optional[str] = None,
        logdir: Optional[str] = None,
        mode: Optional[str] = None,
        tags: Optional[List[str]] = None,
        verbose: int = 0,
        **kwargs: Any,
    )

Source from the content-addressed store, hash-verified

52 """
53
54 def __init__(
55 self,
56 *,
57 project: Optional[str] = None,
58 workspace: Optional[str] = None,
59 experiment_name: Optional[str] = None,
60 description: Optional[str] = None,
61 log_dir: Optional[str] = None,
62 logdir: Optional[str] = None,
63 mode: Optional[str] = None,
64 tags: Optional[List[str]] = None,
65 verbose: int = 0,
66 **kwargs: Any,
67 ) -> None:
68 if logdir is not None:
69 import warnings
70
71 warnings.warn(
72 "The `logdir` parameter is deprecated, use `log_dir` instead.", DeprecationWarning, stacklevel=2
73 )
74 log_dir = logdir
75
76 Callback.__init__(self)
77 _Sb3BaseCallback.__init__(self, verbose=verbose)
78
79 tags = list(tags) if tags else []
80 if "stable_baselines3" not in tags:
81 tags.append("stable_baselines3")
82
83 self._init_kwargs: Dict[str, Any] = {}
84 for key, value in [
85 ("project", project),
86 ("workspace", workspace),
87 ("experiment_name", experiment_name),
88 ("description", description),
89 ("log_dir", log_dir),
90 ("mode", mode),
91 ("tags", tags),
92 ]:
93 if value is not None:
94 self._init_kwargs[key] = value
95 self._init_kwargs.update(kwargs)
96 self._init_kwargs.pop("callbacks", None)
97
98 self._initialized = False
99 self._pending_config: Dict[str, Any] = {}
100
101 @property
102 def name(self) -> str:

Callers

nothing calls this directly

Calls 3

warnMethod · 0.80
popMethod · 0.80
updateMethod · 0.45

Tested by

no test coverage detected