rm newline; update test dtype

coreylowman · Nov 6, 2023 · 1f2b10d · 1f2b10d
1 parent c76ce32
commit 1f2b10d
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/dfdx/src/nn/layers/on.rs b/dfdx/src/nn/layers/on.rs
@@ -44,7 +44,6 @@ mod tests {
         pub b: B,
     }
 
-
     #[input_wrapper]
     pub struct Split1<Forward, Skip> {
         pub forward: Forward,
@@ -74,7 +73,8 @@ mod tests {
     fn test_residual_add_backward() {
         let dev: TestDevice = Default::default();
 
-        let model = dev.build_module::<f32>(<ResidualAdd1<LinearConstConfig<2, 2>>>::default());
+        let model =
+            dev.build_module::<TestDtype>(<ResidualAdd1<LinearConstConfig<2, 2>>>::default());
 
         let x: Tensor<Rank2<4, 2>, f32, _> = dev.sample_normal();
         let x = x.to_dtype::<TestDtype>();
@@ -115,7 +115,8 @@ mod tests {
     fn test_residual_add_backward2() {
         let dev: TestDevice = Default::default();
 
-        let model = dev.build_module::<f32>(<ResidualAdd2<LinearConstConfig<2, 2>>>::default());
+        let model =
+            dev.build_module::<TestDtype>(<ResidualAdd2<LinearConstConfig<2, 2>>>::default());
 
         let x: Tensor<Rank2<4, 2>, f32, _> = dev.sample_normal();
         let x = x.to_dtype::<TestDtype>();