Fixing typing

vasppy/calculation.py

@@ -1,11 +1,11 @@
-import yaml
+import yaml # type: ignore
 import re
 from collections import Counter
 
 
 class Calculation:
     """
-    class describing a single VASP calculation
+    Class describing a single VASP calculation
     """
 
     def __init__(self, title, energy, stoichiometry):

vasppy/data/potcar_data.py

@@ -1,5 +1,5 @@
 import os
-import yaml
+import yaml # type: ignore
 
 my_path = os.path.dirname(__file__)
 

vasppy/doscar.py

@@ -272,6 +272,7 @@ def plot_pdos(
     ) -> Figure:
         if not ax:
             fig, ax = plt.subplots(1, 1, figsize=(8.0, 3.0))
+            assert(isinstance(ax, Axes))
         else:
             fig = None
         if not colors:

vasppy/scripts/vasp_summary.py

@@ -2,7 +2,7 @@
 
 import sys
 from pathlib import Path
-import yaml
+import yaml # type: ignore
 import tqdm  # type: ignore
 from multiprocessing import Pool
 import argparse

vasppy/summary.py

@@ -12,7 +12,7 @@
 from vasppy.data.potcar_data import potcar_md5sum_data
 from vasppy.utils import file_md5, md5sum, match_filename, cd
 from xml.etree import ElementTree as ET
-import yaml
+import yaml # type: ignore
 import glob
 import re
 

vasppy/vaspmeta.py

@@ -1,4 +1,4 @@
-import yaml
+import yaml # type: ignore
 
 
 class VASPMeta: