chore: pre-commit fixes

kraina-ai · Apr 19, 2024 · 9160c38 · 9160c38
1 parent 2bfa5cc
commit 9160c38
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -254,4 +254,5 @@ ignore_packages = [
     'nvidia-cusparse-cu12',     # uses NVIDIA license
     'nvidia-nccl-cu12',         # uses NVIDIA license
     'nvidia-nvtx-cu12',         # uses NVIDIA license
+    'nvidia-nvjitlink-cu12',    # uses NVIDIA license
 ]
diff --git a/srai/datasets/foursquare_checkins/dataset.py b/srai/datasets/foursquare_checkins/dataset.py
@@ -4,6 +4,7 @@
 This module contains Foursquare Checkins Dataset.
 """
 
+import itertools
 from typing import Optional
 
 import geopandas as gpd
@@ -42,7 +43,7 @@ def _preprocessing(
             data=df.drop(columns=["latitude", "longitude"]),
             geometry=df.apply(
                 lambda row: LineString(
-                    [Point(lon, lat) for lon, lat in zip(row["longitude"], row["latitude"])]
+                    list(itertools.starmap(Point, zip(row["longitude"], row["latitude"])))
                 ),
                 axis=1,
             ),

diff --git a/srai/loaders/hf_loader.py b/srai/loaders/hf_loader.py
@@ -47,7 +47,7 @@ def load(self, dataset_name: str, name: Optional[str] = None) -> gpd.GeoDataFram
         from datasets import load_dataset
 
         dataset = load_dataset(
-            f"{dataset_name}", name=name, token=self.hf_token, trust_remote_code=True
+            dataset_name, name=name, token=self.hf_token, trust_remote_code=True
         )  # download dataset from HF
 
         df = dataset["train"].to_pandas()