Fix function parameter order and update test case

PaddlePaddle · Mar 30, 2024 · 33ba331 · 33ba331
1 parent 1d4a2a6
commit 33ba331
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/python/paddle/tensor/linalg.py b/python/paddle/tensor/linalg.py
@@ -2183,7 +2183,7 @@ def bmm(x, y, name=None):
 
 
 def histogram(
-    input, weight=None, density: bool = False, bins=100, min=0, max=0, name=None
+    input, density: bool = False, bins=100, min=0, max=0, weight=None, name=None
 ):
     """
     Computes the histogram of a tensor. The elements are sorted into equal width bins between min and max.

diff --git a/test/legacy_test/test_imperative_layers.py b/test/legacy_test/test_imperative_layers.py
@@ -85,7 +85,9 @@ def test_layer_str(self):
         self.assertEqual(str(module), 'Tanhshrink()')
 
         module = nn.ThresholdedReLU()
-        self.assertEqual(str(module), 'ThresholdedReLU(threshold=1.0)')
+        self.assertEqual(
+            str(module), 'ThresholdedReLU(threshold=1.0, value=0.0)'
+        )
 
         module = nn.LogSigmoid()
         self.assertEqual(str(module), 'LogSigmoid()')