chore: add uuid to uniqueid (#134)

* chore: add uuid to uniqueid
EdanToledo · Feb 17, 2025 · 61c729d · 61c729d
1 parent 56beb38
commit 61c729d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 10 deletions.
diff --git a/plotting/plotting.ipynb b/plotting/plotting.ipynb
@@ -134,7 +134,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "RESULTS_FOLDER_DIR = \"./results\" # Path to the folder containing the results - absolute path might be needed\n",
+    "RESULTS_FOLDER_DIR = \"../results\" # Path to the folder containing the results - absolute path might be needed\n",
     "ENVIRONMENT_SUITE = \"\" # e.g. \"brax\" or \"gymnax\""
    ]
   },
@@ -309,7 +309,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3.8.15 ('emarl-use')",
+   "display_name": "stx",
    "language": "python",
    "name": "python3"
   },
@@ -323,14 +323,9 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.14"
+   "version": "3.10.15"
   },
-  "orig_nbformat": 4,
-  "vscode": {
-   "interpreter": {
-    "hash": "a3f73cd21692be3b017e951052d3ef2e5dfb4cd6dc5b6f5415446a15c3700b84"
-   }
-  }
+  "orig_nbformat": 4
  },
  "nbformat": 4,
  "nbformat_minor": 2

diff --git a/stoix/utils/logger.py b/stoix/utils/logger.py
@@ -5,6 +5,7 @@
 from datetime import datetime
 from enum import Enum
 from typing import Dict, List, Union
+from uuid import uuid4
 
 import jax
 import neptune
@@ -344,7 +345,7 @@ def _make_multi_logger(cfg: DictConfig) -> BaseLogger:
     """Creates a MultiLogger given a config"""
 
     loggers: List[BaseLogger] = []
-    unique_token = datetime.now().strftime("%Y%m%d%H%M%S")
+    unique_token = datetime.now().strftime("%Y%m%d%H%M%S") + str(uuid4())
 
     if (
         (cfg.logger.use_neptune or cfg.logger.use_wandb)