From 2645b907ba6fa677de16aa63c14c557d036c5bea Mon Sep 17 00:00:00 2001
From: jh <jmhummel@gmail.com>
Date: Thu, 11 Mar 2021 23:25:22 -0800
Subject: [PATCH] fix: missing import

---
 mainscripts/Trainer.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py
index fdfe386..4056fe1 100644
--- a/mainscripts/Trainer.py
+++ b/mainscripts/Trainer.py
@@ -4,6 +4,8 @@ import traceback
 import queue
 import threading
 import time
+from enum import Enum
+
 import numpy as np
 import itertools
 from pathlib import Path