From 0dc9544952e69c4b90e8a97fd25c24e1d53d27f3 Mon Sep 17 00:00:00 2001 From: Mayank Mittal Date: Mon, 11 Dec 2023 12:02:31 +0100 Subject: [PATCH] updates to new neptune --- rsl_rl/utils/neptune_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rsl_rl/utils/neptune_utils.py b/rsl_rl/utils/neptune_utils.py index 2b3402d..f06cc62 100644 --- a/rsl_rl/utils/neptune_utils.py +++ b/rsl_rl/utils/neptune_utils.py @@ -8,14 +8,14 @@ from dataclasses import asdict from torch.utils.tensorboard import SummaryWriter try: - import neptune.new as neptune + import neptune except ModuleNotFoundError: raise ModuleNotFoundError("neptune-client is required to log to Neptune.") class NeptuneLogger: def __init__(self, project, token): - self.run = neptune.init(project=project, api_token=token) + self.run = neptune.init_run(project=project, api_token=token) def store_config(self, env_cfg, runner_cfg, alg_cfg, policy_cfg): self.run["runner_cfg"] = runner_cfg