1313
1414import namex
1515
16- package = "keras"
16+ PACKAGE = "keras"
1717BUILD_DIR_NAME = "tmp_build_dir"
1818
1919
@@ -28,7 +28,7 @@ def copy_source_to_build_directory(root_path):
2828 shutil .rmtree (build_dir )
2929 os .mkdir (build_dir )
3030 shutil .copytree (
31- package , os .path .join (build_dir , package ), ignore = ignore_files
31+ PACKAGE , os .path .join (build_dir , PACKAGE ), ignore = ignore_files
3232 )
3333 return build_dir
3434
@@ -163,12 +163,12 @@ def update_package_init(template_fname, dest_fname, api_module):
163163def build ():
164164 # Backup the `keras/__init__.py` and restore it on error in api gen.
165165 root_path = os .path .dirname (os .path .abspath (__file__ ))
166- code_api_dir = os .path .join (root_path , package , "api" )
167- code_init_fname = os .path .join (root_path , package , "__init__.py" )
166+ code_api_dir = os .path .join (root_path , PACKAGE , "api" )
167+ code_init_fname = os .path .join (root_path , PACKAGE , "__init__.py" )
168168 # Create temp build dir
169169 build_dir = copy_source_to_build_directory (root_path )
170- build_api_dir = os .path .join (build_dir , package , "api" )
171- build_init_fname = os .path .join (build_dir , package , "__init__.py" )
170+ build_api_dir = os .path .join (build_dir , PACKAGE , "api" )
171+ build_init_fname = os .path .join (build_dir , PACKAGE , "__init__.py" )
172172 build_api_init_fname = os .path .join (build_api_dir , "__init__.py" )
173173 try :
174174 os .chdir (build_dir )
@@ -184,7 +184,7 @@ def build():
184184 # Add __version__ to `api/`.
185185 export_version_string (build_api_init_fname )
186186 # Creates `_tf_keras` with full keras API
187- create_legacy_directory (package_dir = os .path .join (build_dir , package ))
187+ create_legacy_directory (package_dir = os .path .join (build_dir , PACKAGE ))
188188 # Update toplevel init with all `api/` imports.
189189 api_module = importlib .import_module (f"{ BUILD_DIR_NAME } .keras.api" )
190190 update_package_init (code_init_fname , build_init_fname , api_module )
0 commit comments