Formate code

Author: FlorentinD

examples/kge-predict-transe-pyg-train.ipynb

@@ -411,7 +411,7 @@
     "        print(f\"Epoch: {epoch:03d}, Loss: {loss:.4f}\")\n",
     "        if epoch % 75 == 0:\n",
     "            rank, hits = test(val_tensor_data)\n",
-    "            print(f\"Epoch: {epoch:03d}, Val Mean Rank: {rank:.2f}, \" f\"Val Hits@10: {hits:.4f}\")\n",
+    "            print(f\"Epoch: {epoch:03d}, Val Mean Rank: {rank:.2f}, Val Hits@10: {hits:.4f}\")\n",
     "\n",
     "    torch.save(model, f\"./model_{epoch_count}.pt\")\n",
     "\n",

graphdatascience/graph/graph_entity_ops_runner.py

@@ -158,7 +158,7 @@ def _process_result(
             duplicate_properties = set(db_node_properties).intersection(set(node_properties))
             if duplicate_properties:
                 raise ValueError(
-                    f"Duplicate property keys '{duplicate_properties}' in db_node_properties and " f"node_properties."
+                    f"Duplicate property keys '{duplicate_properties}' in db_node_properties and node_properties."
                 )
 
             unique_node_ids = result["nodeId"].drop_duplicates().tolist()

graphdatascience/query_runner/cypher_graph_constructor.py

@@ -110,9 +110,8 @@ def _should_warn_about_arrow_missing(self) -> bool:
             should_warn = license == "Licensed"
         except Exception as e:
             # It's not a user's concern whether Arrow is set up or not in AuraDS.
-            if (
-                "There is no procedure with the name `gds.debug.sysInfo` "
-                "registered for this database instance." in str(e)
+            if "There is no procedure with the name `gds.debug.sysInfo` registered for this database instance." in str(
+                e
             ):
                 should_warn = False
             else:

graphdatascience/retry_utils/retry_utils.py

@@ -14,7 +14,7 @@ def log_it(retry_state: RetryCallState) -> None:
         if retry_state.attempt_number > 1:  # only log on actual retry
             logger.log(
                 log_level,
-                f"Retry of '{fn_name}', " f"attempt: {retry_state.attempt_number}",
+                f"Retry of '{fn_name}', attempt: {retry_state.attempt_number}",
             )
 
     return log_it