Style fixes.

whitequark · whitequark · commit 0ae13ac1b9e8 · 2015-07-02T22:38:55.000+03:00
diff --git a/artiq/compiler/module.py b/artiq/compiler/module.py
@@ -27,13 +27,13 @@ def __init__(self, source_buffer, engine=None):
         self.globals = asttyped_rewriter.globals
 
     @classmethod
-    def from_string(klass, source_string, name="input.py", first_line=1, engine=None):
-        return klass(source.Buffer(source_string + "\n", name, first_line), engine=engine)
+    def from_string(cls, source_string, name="input.py", first_line=1, engine=None):
+        return cls(source.Buffer(source_string + "\n", name, first_line), engine=engine)
 
     @classmethod
-    def from_filename(klass, filename, engine=None):
+    def from_filename(cls, filename, engine=None):
         with open(filename) as f:
-            return klass(source.Buffer(f.read(), filename, 1), engine=engine)
+            return cls(source.Buffer(f.read(), filename, 1), engine=engine)
 
     def __repr__(self):
         printer = types.TypePrinter()
diff --git a/artiq/compiler/testbench/inferencer.py b/artiq/compiler/testbench/inferencer.py
@@ -42,16 +42,16 @@ def generic_visit(self, node):
                                        ":{}".format(self.type_printer.name(node.type)))
 
 def main():
-    if len(sys.argv) > 1 and sys.argv[1] == '+diag':
+    if len(sys.argv) > 1 and sys.argv[1] == "+diag":
         del sys.argv[1]
         def process_diagnostic(diag):
             print("\n".join(diag.render(only_line=True)))
-            if diag.level == 'fatal':
+            if diag.level == "fatal":
                 exit()
     else:
         def process_diagnostic(diag):
             print("\n".join(diag.render()))
-            if diag.level in ('fatal', 'error'):
+            if diag.level in ("fatal", "error"):
                 exit(1)
 
     engine = diagnostic.Engine()
diff --git a/artiq/compiler/testbench/module.py b/artiq/compiler/testbench/module.py
@@ -3,16 +3,16 @@
 from .. import Module
 
 def main():
-    if len(sys.argv) > 1 and sys.argv[1] == '+diag':
+    if len(sys.argv) > 1 and sys.argv[1] == "+diag":
         del sys.argv[1]
         def process_diagnostic(diag):
             print("\n".join(diag.render(only_line=True)))
-            if diag.level == 'fatal':
+            if diag.level == "fatal":
                 exit()
     else:
         def process_diagnostic(diag):
             print("\n".join(diag.render()))
-            if diag.level in ('fatal', 'error'):
+            if diag.level in ("fatal", "error"):
                 exit(1)
 
     engine = diagnostic.Engine()
