remove unused imports

surfaxe/analysis.py

@@ -1,21 +1,14 @@
 # Pymatgen
-from pymatgen.core import Structure, Element, Lattice
-from pymatgen.core.structure import SiteCollection
+from pymatgen.core import Structure
 from pymatgen.analysis.local_env import CrystalNN, CutOffDictNN
 from pymatgen.io.vasp.outputs import Locpot
 
 # Misc 
-import os
 import math
 import numpy as np
 import pandas as pd
 import warnings
 
-# Matplotlib
-import matplotlib as mpl
-from matplotlib import pyplot as plt
-from mpl_toolkits.axes_grid1 import make_axes_locatable
-
 # surfaxe
 from surfaxe.generation import oxidation_states
 from surfaxe.io import plot_bond_analysis, plot_electrostatic_potential

surfaxe/convergence.py

@@ -1,11 +1,9 @@
 # Pymatgen
-from pymatgen.core import Structure
 from pymatgen.io.vasp.outputs import Vasprun, Outcar
 from pymatgen.analysis.local_env import CrystalNN
 
 # Misc
 import os
-import numpy as np
 import pandas as pd
 import warnings
 import functools

surfaxe/generation.py

@@ -5,7 +5,6 @@
 
 # misc
 import warnings
-import os
 import itertools
 import functools
 import multiprocessing

surfaxe/vasp_data.py

@@ -1,7 +1,5 @@
 # Pymatgen  
-from pymatgen.core.surface import Slab
-from pymatgen.core import Structure, Element
-from pymatgen.core.sites import PeriodicSite
+from pymatgen.core import Structure
 from pymatgen.io.vasp.outputs import Locpot, Outcar, Vasprun
 from pymatgen.analysis.local_env import CrystalNN
 

tests/tests_analysis/test_analysis.py

@@ -1,8 +1,6 @@
 import unittest
-import warnings
 import os
 from pathlib import Path
-from pymatgen.core import Structure
 from surfaxe.analysis import simple_nn, complex_nn, cart_displacements, \
 bond_analysis, electrostatic_potential
 

tests/tests_convergence/test_convergence.py

@@ -1,9 +1,5 @@
 import unittest
-import warnings
-import os
 from pathlib import Path
-from pymatgen.core import Structure
-from pymatgen.core.surface import Slab
 from surfaxe.convergence import parse_fols
 
 fols = str(Path(__file__).parents[2].joinpath('example_data/convergence/Y2Ti2S2O5'))

tests/tests_generation/test_generation.py

@@ -1,9 +1,7 @@
 import unittest
-import warnings
 import os
 import shutil
 from pathlib import Path
-from pymatgen.core import Structure
 from pymatgen.core.surface import Slab
 from surfaxe.generation import get_slabs_max_index, get_slabs_single_hkl
 

tests/tests_io/test_io.py

@@ -1,11 +1,7 @@
 import unittest
-import warnings
-import os
 from pathlib import Path
-from pymatgen.core import Structure
 from pymatgen.core.surface import Slab
 from surfaxe.io import load_config_dict, slab_from_file
-import surfaxe.io 
 
 class LoadTestCase(unittest.TestCase): 
     def setUp(self):