removed obsolete code; added missing __init__ for predictor
This commit is contained in:
parent
5c5d132d7f
commit
4fcd1e79d3
0
image_prediction/predictor/__init__.py
Normal file
0
image_prediction/predictor/__init__.py
Normal file
@ -22,13 +22,6 @@ def estimator(estimator_adapter, classes):
|
||||
return estimator
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def input_output_mapper(input_batch, classes):
|
||||
"""Mocks the internal, real estimator of an EstimatorAdapter object."""
|
||||
outputs = random.choices(range(len(classes)), k=len(input_batch))
|
||||
return outputs
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def estimator_adapter(estimator_type, keras_model, output_batch_generator, monkeypatch):
|
||||
if estimator_type == "mock":
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user