Python formatting

Signed-off-by: Ganesan Ramalingam <grama@microsoft.com>

Author: gramalingam

onnxscript/onnxscript/__init__.py

@@ -6,6 +6,8 @@
 
 # ElementType: an enumeration encoding the allowed element types in a tensor
 # Corresponds to TensorProto::DataType
+
+
 class ElementType:
     UNDEFINED = 0
 

onnxscript/onnxscript/types/__init__.py

@@ -1,17 +1,18 @@
 import onnx
 import onnx.helper
 
+
 class Tensor:
     # Reference implementation placeholder
     # represents a generic ONNX tensor type
-    def __init__(self, dtype = onnx.TensorProto.UNDEFINED, shape = None) -> None:
+    def __init__(self, dtype=onnx.TensorProto.UNDEFINED, shape=None) -> None:
         self.dtype = dtype
         self.shape = shape
-    
+
     def __str__(self) -> str:
         shapestr = str(self.shape) if self.shape else "[...]"
         return onnx.TensorProto.DataType.Name(self.dtype) + shapestr
-    
+
     def to_type_proto(self):
         # TODO: handle None
         return onnx.helper.make_tensor_type_proto(self.dtype, self.shape)
@@ -23,11 +24,12 @@ def to_type_proto(self):
 #    x : FLOAT['M', 'N'] (a tensor of rank 2 of unknown dimensions, with symbolic names)
 #    x : FLOAT[128, 1024] (a tensor of rank 2 of known dimensions)
 
+
 class ParametricTensor:
     def __init__(self, dtype) -> None:
         self.dtype = dtype
-    
-    def __getitem__ (self, shape):
+
+    def __getitem__(self, shape):
         def mk_dim(dim):
             r = onnx.TensorShapeProto.Dimension()
             if (isinstance(dim, int)):
@@ -44,26 +46,27 @@ def mk_dim(dim):
         else:
             s = [shape]
         return Tensor(self.dtype, s)
-    
+
     def to_type_proto(self):
         return onnx.helper.make_tensor_type_proto(self.dtype, ())
-    
+
     def __str__(self) -> str:
         return onnx.TensorProto.DataType.Name(self.dtype)
 
-FLOAT = ParametricTensor (onnx.TensorProto.FLOAT)
-UINT8 = ParametricTensor (onnx.TensorProto.UINT8)
-INT8 = ParametricTensor (onnx.TensorProto.INT8)
-UINT16 = ParametricTensor (onnx.TensorProto.UINT16)
-INT16 = ParametricTensor (onnx.TensorProto.INT16)
-INT32 = ParametricTensor (onnx.TensorProto.INT32)
-INT64 = ParametricTensor (onnx.TensorProto.INT64)
-STRING = ParametricTensor (onnx.TensorProto.STRING)
-BOOL = ParametricTensor (onnx.TensorProto.BOOL)
-FLOAT16 = ParametricTensor (onnx.TensorProto.FLOAT16)
-DOUBLE = ParametricTensor (onnx.TensorProto.DOUBLE)
-UINT32 = ParametricTensor (onnx.TensorProto.UINT32)
-UINT64 = ParametricTensor (onnx.TensorProto.UINT64)
-COMPLEX64 = ParametricTensor (onnx.TensorProto.COMPLEX64)
-COMPLEX128 = ParametricTensor (onnx.TensorProto.COMPLEX128)
-BFLOAT16 = ParametricTensor (onnx.TensorProto.BFLOAT16)
+
+FLOAT = ParametricTensor(onnx.TensorProto.FLOAT)
+UINT8 = ParametricTensor(onnx.TensorProto.UINT8)
+INT8 = ParametricTensor(onnx.TensorProto.INT8)
+UINT16 = ParametricTensor(onnx.TensorProto.UINT16)
+INT16 = ParametricTensor(onnx.TensorProto.INT16)
+INT32 = ParametricTensor(onnx.TensorProto.INT32)
+INT64 = ParametricTensor(onnx.TensorProto.INT64)
+STRING = ParametricTensor(onnx.TensorProto.STRING)
+BOOL = ParametricTensor(onnx.TensorProto.BOOL)
+FLOAT16 = ParametricTensor(onnx.TensorProto.FLOAT16)
+DOUBLE = ParametricTensor(onnx.TensorProto.DOUBLE)
+UINT32 = ParametricTensor(onnx.TensorProto.UINT32)
+UINT64 = ParametricTensor(onnx.TensorProto.UINT64)
+COMPLEX64 = ParametricTensor(onnx.TensorProto.COMPLEX64)
+COMPLEX128 = ParametricTensor(onnx.TensorProto.COMPLEX128)
+BFLOAT16 = ParametricTensor(onnx.TensorProto.BFLOAT16)

onnxscript/test/converter_test.py

@@ -5,12 +5,13 @@
 
 CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
 
+
 class TestConverter(unittest.TestCase):
-    def _convert (self, script):
+    def _convert(self, script):
         converter = Converter()
         converter.convert(script)
 
-    def _convert_and_save (self, script):
+    def _convert_and_save(self, script):
         converter = Converter()
         fnlist = converter.convert(script)
         TEST_OUTPUT_DIR = os.path.join(CURRENT_DIR, "testoutputs")
@@ -37,9 +38,10 @@ def foo(x):
     return msdomain.bar(x, x)
 """
         self._convert(script)
+
     def test_onnxfns(self):
         self._convert(os.path.join(CURRENT_DIR, "onnxfns.py"))
-    
+
     def test_models(self):
         self._convert_and_save(os.path.join(CURRENT_DIR, "onnxmodels.py"))
 
@@ -49,5 +51,6 @@ def test_if_models(self):
     def test_loop_models(self):
         self._convert_and_save(os.path.join(CURRENT_DIR, "loop.py"))
 
+
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()