Fix imports

IBM · Sep 3, 2024 · c1e2c4f · c1e2c4f
1 parent d18d8b4
commit c1e2c4f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/examples/sdk/hello_str.py b/examples/sdk/hello_str.py
@@ -1,6 +1,6 @@
 from pdl.pdl import exec_str
 
-hello = """
+HELLO = """
 document:
 - Hello,
 - model: watsonx/ibm/granite-20b-code-instruct
@@ -13,7 +13,7 @@
 
 
 def main():
-    result = exec_str(hello)
+    result = exec_str(HELLO)
     print(result)
 
 

diff --git a/pdl/pdl.py b/pdl/pdl.py
@@ -7,18 +7,15 @@
 
 from . import pdl_interpreter
 from .pdl_ast import (
-    BlocksType,
-    CallBlock,
     LocationType,
-    Message,
     PdlBlock,
     PdlBlocks,
     Program,
     RoleType,
     ScopeType,
     empty_block_location,
 )
-from .pdl_interpreter import InterpreterState, process_prog, step_call
+from .pdl_interpreter import InterpreterState, process_prog
 from .pdl_parser import parse_file, parse_str