sync

allenai · Dec 10, 2024 · f6bbc78 · f6bbc78
1 parent 7fccfb3
commit f6bbc78
Show file tree

Hide file tree

Showing 6 changed files with 767 additions and 24 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,6 @@
 beaker-py>=1.32
 fastapi>=0.115
+google-cloud-bigtable>=2.18
 interrogate>=1.7
 pydantic>=2.8
 pytest>=8.2

diff --git a/rslp/satlas/__init__.py b/rslp/satlas/__init__.py
@@ -8,14 +8,17 @@
 """
 
 from .job_launcher_worker import launch_workers, write_jobs, write_jobs_for_year_months
-from .postprocess import postprocess_points
+from .postprocess import merge_points, smooth_points
 from .predict_pipeline import predict_multi, predict_pipeline
+from .publish import publish_points
 
 workflows = {
     "predict": predict_pipeline,
     "predict_multi": predict_multi,
     "write_jobs": write_jobs,
     "write_jobs_for_year_months": write_jobs_for_year_months,
     "launch_workers": launch_workers,
-    "postprocess_points": postprocess_points,
+    "merge_points": merge_points,
+    "smooth_points": smooth_points,
+    "publish_points": publish_points,
 }