Minor tweak in nn tests.

Hakuyume · Nov 10, 2020 · 6b39763 · 6b39763
1 parent f3b9390
commit 6b39763
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/nn_tests.rs b/tests/nn_tests.rs
@@ -273,15 +273,14 @@ fn linear_test(linear_config: nn::LinearConfig) {
     // forward test
     let input = Tensor::randint(10, &[batch_dim, input_dim], kind::FLOAT_CPU);
     let expected_var_store_size = if linear_config.bias { 2 } else { 1 };
-    let bias_in_var_store = if linear_config.bias { true } else { false };
 
     let output = linear.forward(&input);
     assert_eq!(output.size(), [batch_dim, output_dim]);
 
     assert_eq!(output.size(), [batch_dim, output_dim]);
     assert_eq!(vs.variables().len(), expected_var_store_size);
     assert!(vs.variables().contains_key("weight"));
-    assert_eq!(vs.variables().contains_key("bias"), bias_in_var_store);
+    assert_eq!(vs.variables().contains_key("bias"), linear_config.bias);
 }
 
 #[test]