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
|
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
|
@pytest.fixture
|
||||||
def estimator_adapter(estimator_type, keras_model, output_batch_generator, monkeypatch):
|
def estimator_adapter(estimator_type, keras_model, output_batch_generator, monkeypatch):
|
||||||
if estimator_type == "mock":
|
if estimator_type == "mock":
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user