Skip to content

Commit

Permalink
Changed dependencies .zip file
Browse files Browse the repository at this point in the history
  • Loading branch information
GriffinBabe committed Jun 12, 2024
1 parent df87509 commit 20746f4
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 4 deletions.
2 changes: 1 addition & 1 deletion scripts/inference/cropland_mapping.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@
"--epsg",
type=int,
default=4326,
help="EPSG code for coordiante reference system.",
help="EPSG code of the input `minx`, `miny`, `maxx`, `maxy` parameters.",
)
parser.add_argument(
"start_date", type=str, help="Starting date for data extraction."
Expand Down
8 changes: 5 additions & 3 deletions src/worldcereal/openeo/feature_extractor.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ class PrestoFeatureExtractor(PatchFeatureExtractor):
"""

PRESTO_MODEL_URL = "https://artifactory.vgt.vito.be/artifactory/auxdata-public/worldcereal-minimal-inference/presto.pt" # NOQA
PRESO_WHL_URL = "https://artifactory.vgt.vito.be/artifactory/auxdata-public/worldcereal/dependencies/presto_worldcereal-0.1.0-temp-py3-none-any.whl"
PRESO_WHL_URL = "https://artifactory.vgt.vito.be/artifactory/auxdata-public/worldcereal/dependencies/presto_worldcereal-0.1.0-py3-none-any.whl"
BASE_URL = "https://s3.waw3-1.cloudferro.com/swift/v1/project_dependencies" # NOQA
DEPENDENCY_NAME = "worldcereal_deps.zip"

Expand Down Expand Up @@ -96,9 +96,11 @@ def execute(self, inarr: xr.DataArray) -> xr.DataArray:
sys.path.append(str(deps_dir))

# Debug, print the dependency directory
self.logger.info(f"Dependency directory: {list(Path(deps_dir).iterdir())}")
self.logger.info("Dependency directory: %s", list(Path(deps_dir).iterdir()))

from presto.inference import get_presto_features
from presto.inference import ( # pylint: disable=import-outside-toplevel
get_presto_features,
)

self.logger.info("Extracting presto features")
features = get_presto_features(inarr, presto_model_url, self.epsg)
Expand Down

0 comments on commit 20746f4

Please sign in to comment.