missed import from prev commit

SheffieldSolar · Jan 26, 2021 · 1bd9b4c · 1bd9b4c
1 parent 6df29d4
commit 1bd9b4c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pvlive_api/pvlive.py b/pvlive_api/pvlive.py
@@ -16,7 +16,7 @@
 import inspect
 import pytz
 import requests
-from numpy import nan
+from numpy import nan, int64
 import pandas as pd
 
 class PVLiveException(Exception):
@@ -45,8 +45,8 @@ def __init__(self, retries=3):
         self.max_range = {"national": timedelta(days=365), "regional": timedelta(days=30)}
         self.retries = retries
         self.ggd_lookup = self._get_ggd_lookup()
-        self.gsp_ids = self.ggd_lookup.gsp_id.dropna().astype(np.int64).unique()
-        self.pes_ids = self.ggd_lookup.pes_id.dropna().astype(np.int64).unique()
+        self.gsp_ids = self.ggd_lookup.gsp_id.dropna().astype(int64).unique()
+        self.pes_ids = self.ggd_lookup.pes_id.dropna().astype(int64).unique()
 
     def _get_ggd_lookup(self):
         """Fetch the GGD lookup from the API and convert to Pandas DataFrame."""