bugfix, wrong exception

mickolaua · Feb 2, 2024 · 0c5fc7d · 0c5fc7d
1 parent 5e1c38b
commit 0c5fc7d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = pygrb_lc
-version = 0.3.7.1
+version = 0.3.7.2
 author = Mozgunov Georgiy
 author_email = georgiy99@bk.ru
 description = Python package for GRB analysis

diff --git a/src/pygrb_lc/light_curves/light_curves.py b/src/pygrb_lc/light_curves/light_curves.py
@@ -9,6 +9,7 @@
 import matplotlib as mpl
 import requests
 import pandas as pd
+import urllib.error
 
 from .transformations import rebin_data
 from ..config import LIGHT_CURVE_SAVE
@@ -536,7 +537,7 @@ def load_fits(url: str, file_extension: str = 'fit', save = True):
                         with open(f"{LIGHT_CURVE_SAVE}{url.split('/')[-1]}_v0{i}.pkl", "wb") as f:
                             pickle.dump(data, f)
                     return data
-                except requests.exceptions.HTTPError:
+                except (urllib.error.HTTPError, requests.exceptions.HTTPError):
                     pass
         raise ValueError(f'No data found for {url}_v0{i}.{file_extension}')