Skip to content

Commit 65e8051

Browse files
committed
Fix linting
1 parent 13dd42c commit 65e8051

2 files changed

Lines changed: 11 additions & 10 deletions

File tree

api_gen.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313

1414
import namex
1515

16-
package = "keras"
16+
PACKAGE = "keras"
1717
BUILD_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):
163163
def 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)

setup.cfg

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,10 +24,11 @@ extend-exclude =
2424
guides,
2525

2626

27-
#imported but unused in __init__.py, that's ok.
2827
per-file-ignores =
29-
# import not used
30-
**/__init__.py:F401
28+
# imported but unused in __init__.py, that's ok.
29+
**/__init__.py:E501,F401
3130
**/random.py:F401
31+
# Lines too long in API files
32+
./keras/api/**/__init__.py:E501,F401
3233

3334
max-line-length = 80

0 commit comments

Comments
 (0)