diff --git a/models/tf.py b/models/tf.py
index 96482dd..84359c4 100644
--- a/models/tf.py
+++ b/models/tf.py
@@ -11,7 +11,6 @@ Export:
 """
 
 import argparse
-import logging
 import sys
 from copy import deepcopy
 from pathlib import Path
diff --git a/utils/loggers/wandb/wandb_utils.py b/utils/loggers/wandb/wandb_utils.py
index 47757dd..a4cbaee 100644
--- a/utils/loggers/wandb/wandb_utils.py
+++ b/utils/loggers/wandb/wandb_utils.py
@@ -7,7 +7,6 @@ from contextlib import contextmanager
 from pathlib import Path
 from typing import Dict
 
-import pkg_resources as pkg
 import yaml
 from tqdm import tqdm