diff --git a/pdebench/data_download/download_direct.py b/pdebench/data_download/download_direct.py
index 35393dd..5cd0ee4 100644
--- a/pdebench/data_download/download_direct.py
+++ b/pdebench/data_download/download_direct.py
@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 import argparse
-import os
 from pathlib import Path
 
 import pandas as pd
diff --git a/pdebench/data_download/visualize_pdes.py b/pdebench/data_download/visualize_pdes.py
index b5fddae..1c75218 100644
--- a/pdebench/data_download/visualize_pdes.py
+++ b/pdebench/data_download/visualize_pdes.py
@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 import argparse
-import os
 from pathlib import Path
 
 import h5py