wehub-resource-sync
17 lines
435 B
Python
17 lines
435 B
Python
from mlflow.data.code_dataset_source import CodeDatasetSource
|
|
|
|
|
|
def test_code_dataset_source_from_path():
|
|
tags = {
|
|
"mlflow_source_type": "NOTEBOOK",
|
|
"mlflow_source_name": "some_random_notebook_path",
|
|
}
|
|
code_datasource = CodeDatasetSource(tags)
|
|
assert code_datasource.to_dict() == {
|
|
"tags": tags,
|
|
}
|
|
|
|
|
|
def test_code_datasource_type():
|
|
assert CodeDatasetSource._get_source_type() == "code"
|