Skip to content

Commit

Permalink
fixed test for valid_month token
Browse files Browse the repository at this point in the history
Butsko Christina committed Oct 11, 2024
1 parent 6c43ceb commit 024ffc3
Showing 1 changed file with 16 additions and 5 deletions.
21 changes: 16 additions & 5 deletions tests/test_dataset.py
Original file line number Diff line number Diff line change
@@ -4,11 +4,22 @@
import numpy as np
import pandas as pd
import torch
from presto.dataops import (NDVI_INDEX, NODATAVALUE, NUM_ORG_BANDS,
NUM_TIMESTEPS, S2_RGB_INDEX, S2_NIR_10m_INDEX)
from presto.dataset import (WorldCerealBase, WorldCerealInferenceDataset,
WorldCerealLabelledDataset,
WorldCerealMaskedDataset, filter_remove_noncrops)

from presto.dataops import (
NDVI_INDEX,
NODATAVALUE,
NUM_ORG_BANDS,
NUM_TIMESTEPS,
S2_RGB_INDEX,
S2_NIR_10m_INDEX,
)
from presto.dataset import (
WorldCerealBase,
WorldCerealInferenceDataset,
WorldCerealLabelledDataset,
WorldCerealMaskedDataset,
filter_remove_noncrops,
)
from presto.eval import WorldCerealEval
from presto.masking import MaskParamsNoDw
from presto.presto import Presto

0 comments on commit 024ffc3

Please sign in to comment.