chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,16 @@
|
||||
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"
|
||||
Reference in New Issue
Block a user