change configuration prefix

milenkovicm · Mar 7, 2024 · 9ddd99c · 9ddd99c
1 parent 5cc5f5b
commit 9ddd99c
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -37,7 +37,7 @@ let sql = r#"
 ctx.sql(sql).await?.show().await?;
 
 // ctx.sql("SET torch.cuda_device = 0").await?;
-ctx.sql("SET torch.device = cpu").await?.show().await?;
+ctx.sql("SET torchfusion.device = cpu").await?.show().await?;
 
 // definition of torch model to use
 let sql = r#"

diff --git a/examples/torchfusion.rs b/examples/torchfusion.rs
@@ -8,8 +8,11 @@ async fn main() -> datafusion::error::Result<()> {
 
     ctx.sql(sql).await?.show().await?;
 
-    // ctx.sql("SET torch.cuda_device = 0").await?;
-    ctx.sql("SET torch.device = cpu").await?.show().await?;
+    // ctx.sql("SET torchfusion.cuda_device = 0").await?;
+    ctx.sql("SET torchfusion.device = cpu")
+        .await?
+        .show()
+        .await?;
 
     // definition of torch model to use
     let sql = r#"

diff --git a/src/config.rs b/src/config.rs
@@ -52,12 +52,12 @@ impl ExtensionOptions for TorchConfig {
     fn entries(&self) -> Vec<datafusion::config::ConfigEntry> {
         vec![
             ConfigEntry {
-                key: "device".into(),
+                key: format!("{}.device", Self::PREFIX),
                 value: Some(format!("{:?}", self.device)),
                 description: "Device to run model on. Valid values 'cpu', 'cuda', 'mps', 'vulkan'. Default: 'cpu' ",
             },
             ConfigEntry {
-                key: "cuda_device".into(),
+                key: format!("{}.cuda_device", Self::PREFIX),
                 value: Some(format!("{}", self.cuda_device)),
                 description: "Cuda device to use. Valid value positive integer. Default: 0",
             },
@@ -83,5 +83,5 @@ impl TorchConfig {
 }
 
 impl ConfigExtension for TorchConfig {
-    const PREFIX: &'static str = "torch";
+    const PREFIX: &'static str = "torchfusion";
 }