Merge pull request #21 from TheMesocarp/publish

change integration tests and publish

Author: FiberedSkies

.github/workflows/rust.yaml

@@ -20,7 +20,7 @@ jobs:
       - uses: actions/checkout@v4
 
       - name: Install Rust
-        uses: dtolnay/rust-toolchain@nightly
+        uses: dtolnay/rust-toolchain@stable
         with:
           components: rustfmt
 
@@ -39,7 +39,7 @@ jobs:
       - uses: actions/checkout@v4
 
       - name: Install Rust
-        uses: dtolnay/rust-toolchain@nightly
+        uses: dtolnay/rust-toolchain@stable
 
       - name: Rust Cache
         uses: Swatinem/rust-cache@v2
@@ -56,7 +56,7 @@ jobs:
       - uses: actions/checkout@v4
 
       - name: Install Rust
-        uses: dtolnay/rust-toolchain@nightly
+        uses: dtolnay/rust-toolchain@stable
         with:
           components: clippy
 
@@ -78,12 +78,61 @@ jobs:
       - uses: actions/checkout@v4
 
       - name: Install Rust
-        uses: dtolnay/rust-toolchain@nightly
+        uses: dtolnay/rust-toolchain@stable
 
       - name: Rust Cache
         uses: Swatinem/rust-cache@v2
         with:
           key: rust/test
 
       - name: Run tests
-        run: cargo test --verbose --workspace
+        run: cargo test --verbose --workspace
+
+  toml-fmt:
+    name: taplo
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      
+      - name: Install Rust
+        uses: dtolnay/rust-toolchain@stable
+
+      - name: Install taplo
+        uses: taiki-e/install-action@cargo-binstall
+        with:
+          tool: taplo-cli
+
+      - name: Rust Cache
+        uses: Swatinem/rust-cache@v2
+        with:
+          key: rust/taplo
+
+      - name: Run TOML fmt
+        run: taplo fmt --check
+
+  # semver:
+  #   name: semver
+  #   runs-on: ubuntu-latest
+  #   continue-on-error: true
+  #   steps:
+  #     - uses: actions/checkout@v4
+  #       with:
+  #         fetch-depth: 0
+
+  #     - name: Install Rust
+  #       uses: dtolnay/rust-toolchain@master
+  #       with:
+  #         toolchain: nightly-2025-05-25
+
+  #     - name: Rust Cache
+  #       uses: Swatinem/rust-cache@v2
+  #       with:
+  #         key: rust/semver
+
+  #     - name: Install cargo-semver-checks
+  #       uses: taiki-e/install-action@cargo-binstall
+  #       with:
+  #         tool: cargo-semver-checks
+
+  #     - name: Check semver
+  #       run: cargo semver-checks check-release

Cargo.toml

@@ -1,12 +1,19 @@
 [package]
+
+authors = ["Mesocarp"]
+description = "A deep learning model for spectral diffusion over k-cells in arbitrary cell complexes, built on `candle`"
+edition = "2021"
+homepage = "https://github.com/TheMesocarp/koho"
+keywords = ["ml", "deep_learning", "spectral_diffusion", "sheaves", "TDA"]
+license = "AGPL-3.0"
 name = "koho"
+readme = "README.md"
+repository = "https://github.com/TheMesocarp/koho"
 version = "0.1.0"
-edition = "2021"
+
 
 [dependencies]
 thiserror = "2.0.12"
-num-complex = "0.4.6"
 rand = "0.9.0"
 candle-core = "0.9.1"
-candle-optimisers = { git = "https://github.com/KGrewal1/optimisers.git", version = "0.9.0" }
 candle-nn = "0.9.1"

src/nn/optim/mod.rs

@@ -1,5 +1,4 @@
 use candle_core::{backprop::GradStore, Result as CandleResult, Var};
-use candle_optimisers::{Decay, Momentum};
 
 pub use crate::nn::optim::adam::Adam;
 pub use crate::nn::optim::rmsprop::RMSprop;
@@ -9,6 +8,23 @@ mod adam;
 mod rmsprop;
 mod sgd;
 
+#[derive(Clone, Copy, Debug, PartialEq, PartialOrd)]
+pub enum Decay {
+    /// Weight decay regularisation to penalise large weights
+    WeightDecay(f64),
+    /// Decoupled weight decay as described in [Decoupled Weight Decay Regularization](https://arxiv.org/abs/1711.05101)
+    DecoupledWeightDecay(f64),
+}
+
+/// Type of momentum to use
+#[derive(Copy, Clone, Debug, PartialEq, PartialOrd)]
+pub enum Momentum {
+    /// classical momentum
+    Classical(f64),
+    /// nesterov momentum
+    Nesterov(f64),
+}
+
 /// Trait for optimizers that work with mutable references to variables
 pub trait Optimizer {
     /// Perform one optimization step using the gradients