From 77751037ca70d8d811bdcc32b89872707d82ab62 Mon Sep 17 00:00:00 2001 From: Ahmed Yarub Hani Al Nuaimi Date: Thu, 9 May 2024 21:49:23 -0300 Subject: [PATCH] Test Signed-off-by: Ahmed Yarub Hani Al Nuaimi --- .github/workflows/test.yml | 22 ---------------------- crates/burn-core/src/nn/attention/mask.rs | 23 ----------------------- 2 files changed, 45 deletions(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index cd7d6c559d..6d892d54c2 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -252,25 +252,3 @@ jobs: with: files: lcov.info token: ${{ secrets.CODECOV_TOKEN }} - - check-typos: - runs-on: ubuntu-22.04 - steps: - - name: checkout - uses: actions/checkout@v4.1.5 - - - name: caching - uses: Swatinem/rust-cache@v2.7.3 - with: - key: ${{ runner.os }}-typos-${{ hashFiles('**/Cargo.toml') }} - prefix-key: "v5-rust" - - - name: Install typos - env: - TYPOS_LINK: https://github.com/crate-ci/typos/releases/download - run: | - curl -L "$TYPOS_LINK/v$TYPOS_VERSION/typos-v$TYPOS_VERSION-x86_64-unknown-linux-musl.tar.gz" | - tar xz -C $HOME/.cargo/bin - - - name: run spelling checks using typos - run: cargo xtask run-checks typos diff --git a/crates/burn-core/src/nn/attention/mask.rs b/crates/burn-core/src/nn/attention/mask.rs index df9a301c1c..b1c277f0ee 100644 --- a/crates/burn-core/src/nn/attention/mask.rs +++ b/crates/burn-core/src/nn/attention/mask.rs @@ -115,27 +115,4 @@ mod tests { ]) ); } - - // #[test] - // fn test_generate_padding_mask() { - // let device = ::Device::default(); - // let tokens = vec![ - // vec![3, 3, 3], - // vec![3, 3, 3], - // vec![3, 3, 3, 4], - // vec![3, 3, 3, 4, 10, 15], - // ]; - // - // let mask = generate_padding_mask::(0, tokens, None, &device); - // - // assert_eq!( - // mask.mask.into_data(), - // Data::from([ - // [false, false, false, true, true, true], - // [false, false, false, true, true, true], - // [false, false, false, false, true, true], - // [false, false, false, false, false, false], - // ]) - // ); - // } }