From 4f8bf389d8a6662faf74d0503a3de8f980483311 Mon Sep 17 00:00:00 2001
From: Godwin H <51889272+Godwinh19@users.noreply.github.com>
Date: Sat, 1 Apr 2023 02:13:17 +0100
Subject: [PATCH] Update test_optimizer.py

---
 optimizers/test_optimizer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/optimizers/test_optimizer.py b/optimizers/test_optimizer.py
index 5ef1107..e396c41 100644
--- a/optimizers/test_optimizer.py
+++ b/optimizers/test_optimizer.py
@@ -1,7 +1,7 @@
 import unittest
 import torch
 from tml.optimizers.config import LearningRate, OptimizerConfig
-from .optimizer import compute_lr, LRShim, get_optimizer_class, build_optimizer
+from optimizer import compute_lr, LRShim, get_optimizer_class, build_optimizer
 
 
 class TestComputeLR(unittest.TestCase):