diff --git a/project_1.py b/project_1.py
index d13455a..cfe9c9b 100644
--- a/project_1.py
+++ b/project_1.py
@@ -6,6 +6,7 @@
 """
 
 import os
+import logging
 import numpy as np
 import pandas as pd
 import torch