Minor change to imports

diff --git a/ndcode/pitree/cli.py b/ndcode/pitree/cli.py
index aca10b3..b3fead9 100755 (executable)
--- a/ndcode/pitree/cli.py
+++ b/ndcode/pitree/cli.py
@@ -20,8 +20,8 @@ import getopt
 import os
 import sys
 from ndcode.pitree import element
-from ndcode.pitree import generate_c
-from ndcode.pitree import generate_py
+from ndcode.pitree.generate_c import generate_c
+from ndcode.pitree.generate_py import generate_py
 from ndcode.pitree import lex_yy
 from ndcode.pitree import t_def
 from ndcode.pitree import y_tab
@@ -85,7 +85,7 @@ def main():
   ast.post_process()
   #element.serialize(ast, 'b.xml', 'utf-8')
   #ast = element.deserialize('b.xml', t_def.factory, 'utf-8')
-  (generate_py.generate_py if python else generate_c.generate_c)(
+  (generate_py if python else generate_c)(
     ast,
     home_dir,
     skel_file,