From 76894eff040d86fa3ab8a9ed5b4402c13077b518 Mon Sep 17 00:00:00 2001
From: nathaniel <nathaniel.simard.42@gmail.com>
Date: Wed, 6 Sep 2023 10:25:37 -0400
Subject: [PATCH] Remove candle from release

---
 .github/workflows/publish.yml | 11 -----------
 burn-core/Cargo.toml          |  3 ---
 burn-core/src/backend.rs      | 11 -----------
 burn-core/src/lib.rs          | 10 +---------
 burn/Cargo.toml               |  1 -
 xtask/src/runchecks.rs        |  3 ---
 6 files changed, 1 insertion(+), 38 deletions(-)

diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
index 16f1d3bcde..6cd4d58d18 100644
--- a/.github/workflows/publish.yml
+++ b/.github/workflows/publish.yml
@@ -78,16 +78,6 @@ jobs:
       crate: burn-wgpu
     secrets: inherit
 
-  publish-burn-candle:
-    uses: burn-rs/burn/.github/workflows/publish-template.yml@main
-    needs:
-      - publish-burn-tensor
-      - publish-burn-autodiff
-      - publish-burn-tch
-    with:
-      crate: burn-candle
-    secrets: inherit
-
   publish-burn-core:
     uses: burn-rs/burn/.github/workflows/publish-template.yml@main
     needs:
@@ -96,7 +86,6 @@ jobs:
       - publish-burn-derive
       - publish-burn-tensor
       - publish-burn-autodiff
-      - publish-burn-candle
       - publish-burn-wgpu
       - publish-burn-tch
       - publish-burn-ndarray
diff --git a/burn-core/Cargo.toml b/burn-core/Cargo.toml
index 48e43bf1ab..a5b00488f9 100644
--- a/burn-core/Cargo.toml
+++ b/burn-core/Cargo.toml
@@ -44,14 +44,12 @@ wgpu = ["burn-wgpu"]
 wgpu-autotune = ["wgpu", "burn-wgpu/autotune"]
 
 tch = ["burn-tch"]
-candle = ["burn-candle"]
 
 # Serialization formats
 experimental-named-tensor = ["burn-tensor/experimental-named-tensor"]
 
 test-tch = ["tch"]    # To use tch during testing, default uses ndarray.
 test-wgpu = ["wgpu"]   # To use wgpu during testing, default uses ndarray.
-test-candle = ["candle"] # To use candle during testing, default uses ndarray.
 
 [dependencies]
 
@@ -67,7 +65,6 @@ burn-ndarray = { path = "../burn-ndarray", version = "0.9.0", optional = true, d
 burn-autodiff = { path = "../burn-autodiff", version = "0.9.0", optional = true }
 burn-wgpu = { path = "../burn-wgpu", version = "0.9.0", optional = true }
 burn-tch = { path = "../burn-tch", version = "0.9.0", optional = true }
-burn-candle = { path = "../burn-candle", version = "0.9.0", optional = true }
 
 derive-new = { workspace = true }
 libm = { workspace = true }
diff --git a/burn-core/src/backend.rs b/burn-core/src/backend.rs
index 16ded2a9b4..8bf5ef1bfc 100644
--- a/burn-core/src/backend.rs
+++ b/burn-core/src/backend.rs
@@ -28,14 +28,3 @@ pub type TchBackend<F = f32> = tch::TchBackend<F>;
 /// A TchBackend with autodiffing enabled.
 #[cfg(all(feature = "tch", feature = "autodiff"))]
 pub type TchAutodiffBackend<F = f32> = crate::autodiff::ADBackendDecorator<TchBackend<F>>;
-
-/// Candle module.
-#[cfg(feature = "candle")]
-pub use burn_candle as candle;
-/// A CandleBackend with a default type of f32/i32.
-#[cfg(feature = "candle")]
-pub type CandleBackend<F = f32, I = i32> = candle::CandleBackend<F, I>;
-/// A CandleBackend with autodiffing enabled.
-#[cfg(all(feature = "candle", feature = "autodiff"))]
-pub type CandleAutodiffBackend<F = f32, I = i32> =
-    crate::autodiff::ADBackendDecorator<CandleBackend<F, I>>;
diff --git a/burn-core/src/lib.rs b/burn-core/src/lib.rs
index 73c58f648c..b8a48f52d4 100644
--- a/burn-core/src/lib.rs
+++ b/burn-core/src/lib.rs
@@ -45,20 +45,12 @@ pub mod backend;
 
 extern crate alloc;
 
-#[cfg(all(
-    test,
-    not(feature = "test-tch"),
-    not(feature = "test-wgpu"),
-    not(feature = "test-candle")
-))]
+#[cfg(all(test, not(feature = "test-tch"), not(feature = "test-wgpu"),))]
 pub type TestBackend = burn_ndarray::NdArrayBackend<f32>;
 
 #[cfg(all(test, feature = "test-tch"))]
 pub type TestBackend = burn_tch::TchBackend<f32>;
 
-#[cfg(all(test, feature = "test-candle"))]
-pub type TestBackend = burn_candle::CandleBackend<f32, u32>;
-
 #[cfg(all(test, feature = "test-wgpu", not(target_os = "macos")))]
 pub type TestBackend = burn_wgpu::WgpuBackend<burn_wgpu::Vulkan, f32, i32>;
 
diff --git a/burn/Cargo.toml b/burn/Cargo.toml
index ce1b798e8d..10920d3e00 100644
--- a/burn/Cargo.toml
+++ b/burn/Cargo.toml
@@ -47,7 +47,6 @@ wgpu = ["burn-core/wgpu"]
 wgpu-autotune = ["burn-core/wgpu-autotune"]
 
 tch = ["burn-core/tch"]
-candle = ["burn-core/candle"]
 
 # Experimental
 experimental-named-tensor = ["burn-core/experimental-named-tensor"]
diff --git a/xtask/src/runchecks.rs b/xtask/src/runchecks.rs
index 27def0a0ca..e6a843b537 100644
--- a/xtask/src/runchecks.rs
+++ b/xtask/src/runchecks.rs
@@ -208,9 +208,6 @@ fn burn_core_std() {
     // Run cargo test --features test-tch
     cargo_test(&["-p", "burn-core", "--features", "test-tch"]);
 
-    // Run cargo test --features test-candle
-    // cargo_test(&["-p", "burn-core", "--features", "test-candle"]);
-
     // Run cargo test --features test-wgpu
     cargo_test(&["-p", "burn-core", "--features", "test-wgpu"]);
 }