add deps

Pometry · Jun 20, 2024 · 1323b32 · 1323b32
1 parent 96267dc
commit 1323b32
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/raphtory/Cargo.toml b/raphtory/Cargo.toml
@@ -104,12 +104,11 @@ io = [
 # Enables generating the pyo3 python bindings
 python = [
     "io",
+    "arrow",
     "dep:pyo3",
     "dep:num",
     "dep:display-error-chain",
-    "dep:polars-arrow",
     "polars-arrow?/compute",
-    "dep:polars-parquet",
     "dep:kdam",
     "dep:rpds",
 ]
@@ -119,10 +118,9 @@ search = ["dep:tantivy"]
 vectors = ["dep:futures-util", "dep:async-trait", "dep:async-openai"]
 # storage
 storage = [
+    "arrow",
     "pometry-storage",
-    "dep:polars-arrow",
     "dep:polars-utils",
-    "dep:polars-parquet",
     "dep:memmap2",
     "dep:ahash",
     "dep:tempfile",
@@ -133,3 +131,7 @@ storage = [
     "polars-arrow?/arrow_rs",
     "polars-parquet?/compression",
 ]
+arrow = [
+    "dep:polars-arrow",
+    "dep:polars-parquet",
+]
diff --git a/raphtory/src/core/utils/errors.rs b/raphtory/src/core/utils/errors.rs
@@ -1,4 +1,5 @@
 use crate::core::{utils::time::error::ParseTimeError, ArcStr, Prop, PropType};
+#[cfg(feature = "arrow")]
 use polars_arrow::legacy::error;
 #[cfg(feature = "search")]
 use tantivy;
@@ -7,6 +8,7 @@ use tantivy::query::QueryParserError;
 
 #[derive(thiserror::Error, Debug)]
 pub enum GraphError {
+    #[cfg(feature = "arrow")]
     #[error("Arrow error: {0}")]
     Arrow(#[from] error::PolarsError),
     #[error("Graph error occurred")]

diff --git a/raphtory/src/python/graph/io/parquet_loaders.rs b/raphtory/src/python/graph/io/parquet_loaders.rs
@@ -1,6 +1,6 @@
 use crate::{
     core::{entities::graph::tgraph::InternalGraph, utils::errors::GraphError, Prop},
-    python::graph::io::{dataframe::*, df_loaders::*, panda_loaders::process_pandas_py_df},
+    python::graph::io::{dataframe::*, df_loaders::*},
 };
 use itertools::Itertools;
 use polars_arrow::{
@@ -13,7 +13,6 @@ use polars_parquet::{
     read,
     read::{read_metadata, FileMetaData, FileReader},
 };
-use pyo3::{types::IntoPyDict, PyAny, PyErr, Python};
 use std::{collections::HashMap, path::Path};
 
 pub fn load_nodes_from_parquet(