Merge pull request #9 from pyiron/add_black

Add black
pyiron · Oct 5, 2024 · 48022a0 · 48022a0
2 parents f6ad48b + a6d8534
commit 48022a0
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 0 deletions.
diff --git a/.binder/environment.yml b/.binder/environment.yml
@@ -4,3 +4,4 @@ dependencies:
 - coveralls
 - coverage
 - numpy =1.26.0
+- black =24.8.0
diff --git a/.ci_support/environment.yml b/.ci_support/environment.yml
@@ -4,3 +4,4 @@ dependencies:
 - coveralls
 - coverage
 - numpy =1.26.0
+- black =24.8.0
diff --git a/docs/environment.yml b/docs/environment.yml
@@ -10,3 +10,4 @@ dependencies:
 - coveralls
 - coverage
 - numpy =1.26.0
+- black =24.8.0
diff --git a/stinx/src/generator.py b/stinx/src/generator.py
@@ -2,6 +2,7 @@
 import numpy as np
 import yaml
 import os
+from black import format_str, FileMode
 
 
 indent = 4 * " "
@@ -203,6 +204,7 @@ def export_class(yml_file_name="input_data.yml", py_file_name="input.py"):
         "from stinx.toolkit import fill_values",
     ]
     file_content = "\n".join(imports) + "\n\n\n" + file_content
+    file_content = format_str(file_content, mode=FileMode())
     with open(
         os.path.join(os.path.dirname(__file__), "..", py_file_name), "w"
     ) as f: