fix param order

arangoml · Oct 15, 2023 · a983de9 · a983de9
1 parent 48160f1
commit a983de9
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/adbpyg_adapter/adapter.py b/adbpyg_adapter/adapter.py
@@ -748,13 +748,13 @@ def __process_adb_cursor(
         :type progress_color: str
         :param cursor: The ArangoDB cursor for the current **col**.
         :type cursor: arango.cursor.Cursor
+        :param col_size: The size of **col**.
+        :type col_size: int
         :param process_adb_df: The function to process the cursor data
             (in the form of a Dataframe).
         :type process_adb_df: Callable
         :param col: The ArangoDB collection for the current **cursor**.
         :type col: str
-        :param col_size: The size of **col**.
-        :type col_size: int
         :param adb_map: The ArangoDB -> PyG map.
         :type adb_map: adbpyg_adapter.typings.ADBMap
         :param meta: The metagraph for the current **col**.
@@ -1211,9 +1211,9 @@ def __process_pyg_node_batch(
             meta,
             node_data,
             node_data.num_nodes,
+            is_explicit_metagraph,
             start_index,
             end_index,
-            is_explicit_metagraph,
         )
 
         # 2. Update the PyG Map
@@ -1278,9 +1278,9 @@ def __process_pyg_edge_batch(
             meta,
             edge_data,
             edge_data.num_edges,
+            is_explicit_metagraph,
             start_index,
             end_index,
-            is_explicit_metagraph,
         )
 
         # 3. Set the _from column
@@ -1310,9 +1310,9 @@ def __set_adb_data(
         meta: Union[Set[str], Dict[Any, PyGMetagraphValues]],
         pyg_data: Union[Data, NodeStorage, EdgeStorage],
         pyg_data_size: int,
+        is_explicit_metagraph: bool,
         start_index: int,
         end_index: int,
-        is_explicit_metagraph: bool,
     ) -> DataFrame:
         """PyG -> ArangoDB: A helper method to build the ArangoDB Dataframe for
         the given collection. Is responsible for creating "sub-DataFrames"
@@ -1332,13 +1332,13 @@ def __set_adb_data(
         :param pyg_data_size: The size of the NodeStorage or EdgeStorage of the
             current PyG node or edge type.
         :type pyg_data_size: int
+        :param is_explicit_metagraph: Take the metagraph at face value or not.
+        :type is_explicit_metagraph: bool
         :param start_index: The starting index of the current batch to process.
         :type start_index: int
         :param end_index: The ending index of the current batch to process.
         :type end_index: int
         :type pyg_data: torch_geometric.data.storage.(NodeStorage | EdgeStorage)
-        :param is_explicit_metagraph: Take the metagraph at face value or not.
-        :type is_explicit_metagraph: bool
         :return: The completed DataFrame for the (soon-to-be) ArangoDB collection.
         :rtype: pandas.DataFrame
         :raise ValueError: If an unsupported PyG data value is found.