import os

from bonobo import Graph


def extract():
    env_test_user = os.getenv('ENV_TEST_USER')
    env_test_number = os.getenv('ENV_TEST_NUMBER')
    env_test_string = os.getenv('ENV_TEST_STRING')
    return env_test_user, env_test_number, env_test_string


def load(s: str):
    print(s)


graph = Graph(extract, load)

if __name__ == '__main__':
    from bonobo import run

    run(graph)

@@ -8,32 +8,21 @@
 os%0A%0A
-from
+import
  bonobo
- import Graph
 %0A%0A%0Ad
@@ -286,16 +286,23 @@
 graph = 
+bonobo.
 Graph(ex
@@ -350,36 +350,15 @@
     
-from 
 bonobo
- import run%0A%0A    
+.
 run(

Change import style in example.