Do not use unix only paths in keras/integration_test.py
All python code needs to be able to work on all operating systems, including windows. PiperOrigin-RevId: 232317348
This commit is contained in:
parent
a127a0b5ce
commit
63a13795c2
@ -18,6 +18,8 @@ from __future__ import absolute_import
|
||||
from __future__ import division
|
||||
from __future__ import print_function
|
||||
|
||||
import os
|
||||
|
||||
import numpy as np
|
||||
|
||||
from tensorflow.python import keras
|
||||
@ -236,7 +238,7 @@ class ActivationV2IntegrationTest(keras_parameterized.TestCase):
|
||||
verbose=2)
|
||||
|
||||
output_path = keras.saving.saved_model.export(
|
||||
model, '/tmp/tf_keras_saved_model')
|
||||
model, os.path.join(self.get_temp_dir(), 'tf_keras_saved_model'))
|
||||
loaded_model = keras.saving.saved_model.load_from_saved_model(output_path)
|
||||
self.assertEqual(model.summary(), loaded_model.summary())
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user