format

Author: andrej

operators/softmax/op.py

@@ -21,12 +21,7 @@
 class AIESoftmax(AIEOperatorBase):
 
     def __init__(
-        self, 
-        rows: int, 
-        cols: int, 
-        num_aie_columns=1, 
-        num_channels=1, 
-        context=None
+        self, rows: int, cols: int, num_aie_columns=1, num_channels=1, context=None
     ):
         self.size = rows * cols
         self.rows = rows

operators/softmax/test.py

@@ -84,7 +84,7 @@ def test_softmax(input_length, num_aie_columns, num_channels, tile_size, aie_con
         cols=cols,
         num_aie_columns=num_aie_columns,
         num_channels=num_channels,
-        context=aie_context
+        context=aie_context,
     )
 
     input_buffers = {"in": golden_ref["input"]}