diff --git a/scripts/training/train.py b/scripts/training/train.py
index 09d5d8e..b48e733 100644
--- a/scripts/training/train.py
+++ b/scripts/training/train.py
@@ -4,6 +4,7 @@
 import ast
 import logging
 import os
+import uuid
 import re
 import sys
 import json