Skip to content

Commit

Permalink
Revert: Workaround keras-team/keras#19875
Browse files Browse the repository at this point in the history
  • Loading branch information
lutzroeder committed Jun 22, 2024
1 parent dc85d98 commit 02bf68b
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions tools/keras_script.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@
import json
import os
import pydoc
import pathlib
import re

os.environ["KERAS_BACKEND"] = "jax"
Expand Down Expand Up @@ -170,11 +169,6 @@ def _update_headers(schema, docstring):

def _metadata():

file = pathlib.Path('third_party/env/keras/lib/python3.12/site-packages/keras/__init__.py')
content = file.read_text(encoding='utf-8')
content = content.replace('from keras.api import _tf_keras', 'from keras import _tf_keras')
file.write_text(content, encoding='utf-8')

root = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
json_path = os.path.join(root, 'source', 'keras-metadata.json')
json_root = json.loads(_read(json_path))
Expand Down

0 comments on commit 02bf68b

Please sign in to comment.