fix(convert): keep classes

crunchdao · May 11, 2023 · 2d5c8e0 · 2d5c8e0
1 parent 097ae2f
commit 2d5c8e0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/crunch/__version__.py b/crunch/__version__.py
@@ -1,6 +1,6 @@
 __title__ = 'crunch-cli'
 __description__ = 'crunch-cli - CLI of the CrunchDAO Platform'
-__version__ = '1.1.0'
+__version__ = '1.1.1'
 __author__ = 'Enzo CACERES'
 __author_email__ = '[email protected]'
 __url__ = 'https://github.com/crunchdao/crunch-cli'
diff --git a/crunch/command/convert.py b/crunch/command/convert.py
@@ -39,7 +39,7 @@ def log(message):
         tree = ast.parse(source)
 
         for node in tree.body:
-            if isinstance(node, (ast.Import, ast.ImportFrom, ast.FunctionDef)):
+            if isinstance(node, (ast.Import, ast.ImportFrom, ast.FunctionDef, ast.ClassDef)):
                 valid += 1
                 module.body.append(node)