diff --git a/pyTigerGraph/gds/featurizer.py b/pyTigerGraph/gds/featurizer.py index 6ada9c70..d5fb0aa2 100644 --- a/pyTigerGraph/gds/featurizer.py +++ b/pyTigerGraph/gds/featurizer.py @@ -180,7 +180,7 @@ def _get_db_version(self) -> Tuple[str, str, str]: def _get_algo_dict(self, manifest_file: str) -> dict: # Get algo dict from manifest if manifest_file.startswith("http"): - resp = requests.get(manifest_file) + resp = requests.get(manifest_file, allow_redirects=False, timeout=10) resp.raise_for_status() algo_dict = resp.json() else: @@ -259,7 +259,7 @@ def _install_query_file( """ # Read in the query if query_path.startswith("http"): - resp = requests.get(query_path) + resp = requests.get(query_path, allow_redirects=False, timeout=10) resp.raise_for_status() query = resp.text else: @@ -404,7 +404,7 @@ def _get_query(self, query_name: str) -> str: raise ValueError("Cannot find {} in the library.".format(query_name)) query_path = self.algo_paths[query_name][-1] if query_path.startswith("http"): - resp = requests.get(query_path) + resp = requests.get(query_path, allow_redirects=False, timeout=10) resp.raise_for_status() query = resp.text else: