Minor fix

Author: FilippoOlivo

pina/data/data_module.py

@@ -28,8 +28,10 @@ def __init__(self, dataset):
         have the following cases:
 
         - **Distributed Environment** (multiple GPUs):
-            - Divides the dataset across processes using the rank and world size.
-            - Fetches only the portion of data corresponding to the current process.
+            - Divides the dataset across processes using the rank and world
+                size.
+            - Fetches only the portion of data corresponding to the current
+                process.
         - **Non-Distributed Environment** (single GPU):
             - Fetches the entire dataset.
 
@@ -80,8 +82,8 @@ def __init__(
         Initialize the object, setting the collate function based on whether
         automatic batching is enabled or not.
 
-        :param dict max_conditions_lengths: dict containing the maximum number of
-            data points to consider in a single batch for each condition.
+        :param dict max_conditions_lengths: dict containing the maximum number
+            of data points to consider in a single batch for each condition.
         :param PinaDataset dataset: The dataset where the data is stored.
         """
 

pina/data/dataset.py

@@ -197,7 +197,6 @@ def _retrive_data(self, data, idx_list):
         :return: Dictionary containing the data at the given indices
         :rtype: dict
         """
-        pass
 
 
 class PinaTensorDataset(PinaDataset):

pina/graph.py

@@ -217,7 +217,8 @@ def __new__(
         **kwargs,
     ):
         """
-        Compute the edge attributes and create a new instance of the Graph class.
+        Compute the edge attributes and create a new instance of the Graph
+        class.
 
         :param pos: A tensor of shape (N, D) representing the positions of N
             points in D-dimensional space.