mirror of
https://github.com/wassname/pytorch-ts.git
synced 2026-09-13 13:02:46 +08:00
fix some imports
This commit is contained in:
@@ -0,0 +1,15 @@
|
||||
from pts.dataset import FieldName
|
||||
|
||||
def test_dataset_fields():
|
||||
assert (
|
||||
"feat_static_cat" == FieldName.FEAT_STATIC_CAT
|
||||
), "Error in the FieldName 'feat_static_cat'."
|
||||
assert (
|
||||
"feat_static_real" == FieldName.FEAT_STATIC_REAL
|
||||
), "Error in the FieldName 'feat_static_real'."
|
||||
assert (
|
||||
"feat_dynamic_cat" == FieldName.FEAT_DYNAMIC_CAT
|
||||
), "Error in the FieldName 'feat_dynamic_cat'."
|
||||
assert (
|
||||
"feat_dynamic_real" == FieldName.FEAT_DYNAMIC_REAL
|
||||
), "Error in the FieldName 'feat_dynamic_real'."
|
||||
Reference in New Issue
Block a user