Skip to content

Commit 0dc9544

Browse files
committed
updates to new neptune
1 parent cde1e87 commit 0dc9544

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

rsl_rl/utils/neptune_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,14 +8,14 @@
88
from torch.utils.tensorboard import SummaryWriter
99

1010
try:
11-
import neptune.new as neptune
11+
import neptune
1212
except ModuleNotFoundError:
1313
raise ModuleNotFoundError("neptune-client is required to log to Neptune.")
1414

1515

1616
class NeptuneLogger:
1717
def __init__(self, project, token):
18-
self.run = neptune.init(project=project, api_token=token)
18+
self.run = neptune.init_run(project=project, api_token=token)
1919

2020
def store_config(self, env_cfg, runner_cfg, alg_cfg, policy_cfg):
2121
self.run["runner_cfg"] = runner_cfg

0 commit comments

Comments
 (0)