wehub-resource-sync
593b94c120
pytest / Unit Tests (push) Has been cancelled
pytest / Integration (integration_tests_a) (push) Has been cancelled
pytest / Integration (integration_tests_b) (push) Has been cancelled
pytest / Integration (integration_tests_c) (push) Has been cancelled
pytest / Integration (integration_tests_d) (push) Has been cancelled
pytest / Integration (integration_tests_e) (push) Has been cancelled
pytest / Integration (integration_tests_f) (push) Has been cancelled
pytest / Integration (integration_tests_g) (push) Has been cancelled
pytest / Integration (integration_tests_h) (push) Has been cancelled
pytest / Integration (integration_tests_i) (push) Has been cancelled
pytest / Integration (integration_tests_j) (push) Has been cancelled
pytest / Distributed (distributed_a) (push) Has been cancelled
pytest / Distributed (distributed_b) (push) Has been cancelled
pytest / Distributed (distributed_c) (push) Has been cancelled
pytest / Distributed (distributed_d) (push) Has been cancelled
pytest / Distributed (distributed_e) (push) Has been cancelled
pytest / Distributed (distributed_f) (push) Has been cancelled
pytest / Minimal Install (push) Has been cancelled
pytest / Event File (push) Has been cancelled
pytest (slow) / py-slow (push) Has been cancelled
Publish JSON Schema / publish-schema (push) Has been cancelled
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
import argparse
|
|
import os
|
|
import sys
|
|
import tempfile
|
|
from unittest.mock import Mock
|
|
|
|
# Comet must be imported before the libraries it wraps
|
|
import aim # noqa
|
|
|
|
from ludwig.contribs.aim import AimCallback
|
|
from tests.integration_tests.utils import category_feature, generate_data, image_feature, run_experiment
|
|
|
|
PATH_HERE = os.path.abspath(os.path.dirname(__file__))
|
|
PATH_ROOT = os.path.join(PATH_HERE, "..", "..", "..")
|
|
sys.path.insert(0, os.path.abspath(PATH_ROOT))
|
|
|
|
|
|
def run(csv_filename):
|
|
callback = AimCallback()
|
|
|
|
# Wrap these methods so we can check that they were called
|
|
callback.on_train_init = Mock(side_effect=callback.on_train_init)
|
|
callback.on_train_start = Mock(side_effect=callback.on_train_start)
|
|
|
|
# Image Inputs
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
image_dest_folder = os.path.join(tmpdir, "generated_images")
|
|
|
|
# Inputs & Outputs
|
|
input_features = [image_feature(folder=image_dest_folder)]
|
|
output_features = [category_feature(output_feature=True)]
|
|
rel_path = generate_data(input_features, output_features, csv_filename)
|
|
|
|
# Run experiment
|
|
run_experiment(input_features, output_features, dataset=rel_path, callbacks=[callback])
|
|
|
|
# Check that these methods were called at least once
|
|
callback.on_train_init.assert_called()
|
|
callback.on_train_start.assert_called()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--csv-filename", required=True)
|
|
args = parser.parse_args()
|
|
run(args.csv_filename)
|