remove keck drp framework

pyproject.toml

@@ -11,8 +11,7 @@ dependencies = [
   "astropy",
   "numpy",
   "matplotlib",
-  "scipy",
-  "keckdrpframework"
+  "scipy"
 ]
 requires-python = ">=3.9"
 authors = [

src/main.py

@@ -6,7 +6,6 @@
 import yaml
 import os
 import subprocess
-import time
 from workflows.flows.batch_flat_flow import batch_process_all_flats
 
 def load_config(config_path="config/config.yaml"):

src/workflows/prefect_tasks/create_master_flat.py

@@ -1,7 +1,5 @@
 from prefect import task
-from keckdrpframework.models.arguments import Arguments
-from keck_primitives.create_master_flat import CreateMasterFlat
-from keck_primitives.utils import DummyAction, DummyContext
+from core.flat import create_master_flat
 
 
 @task(name="Create Master Flat")
@@ -27,19 +25,12 @@ def create_master_flat_task(
     Returns:
         Master flat correction array (multiply science data by this)
     """
-    args = Arguments()
-    args["flat_data"] = flat_data
-    args["method"] = "spectroscopic"
-
-    if slit_positions is not None:
-        args["slit_positions"] = slit_positions
-    args["slit_width"] = slit_width
-    args["n_knots_spectral"] = n_knots_spectral
-    args["low_signal_threshold"] = low_signal_threshold
-    args["edge_trim_pixels"] = edge_trim_pixels
-
-    action = DummyAction(args=args)
-    context = DummyContext()
-
-    result = CreateMasterFlat(action, context)._perform(args, config={})
-    return result["correction"]
+    correction = create_master_flat(
+        flat_data,
+        slit_positions=slit_positions,
+        slit_width=slit_width,
+        n_knots_spectral=n_knots_spectral,
+        low_signal_threshold=low_signal_threshold,
+        edge_trim_pixels=edge_trim_pixels
+    )
+    return correction

src/workflows/prefect_tasks/load_flat.py

@@ -1,16 +1,8 @@
 from prefect import task
-from keckdrpframework.models.arguments import Arguments
-from keck_primitives.load_flat import LoadFlat
-from keck_primitives.utils import DummyAction, DummyContext
+from core.flat import load_flat_frame
 
 
 @task(name="Load Flat Frame")
 def load_flat_frame_task(filepath: str):
-    args = Arguments()
-    args["filepath"] = filepath
-
-    action = DummyAction(args=args)
-    context = DummyContext()
-
-    result = LoadFlat(action, context)._perform(args, config={})
-    return result["flat_data"], result["header"]
+    flat_data, header = load_flat_frame(filepath)
+    return flat_data, header

src/workflows/prefect_tasks/qa_plot.py

@@ -1,18 +1,8 @@
 from prefect import task
-from keckdrpframework.models.arguments import Arguments
-from keck_primitives.qa_plot import GenerateQAPlot
-from keck_primitives.utils import DummyAction, DummyContext
+from core.qa import generate_qa_plot
 
 
 @task(name="Generate QA Plot")
 def generate_qa_plot_task(data, output_path: str, title: str = "Flat QA"):
-    args = Arguments()
-    args["data"] = data
-    args["output_path"] = output_path
-    args["title"] = title
-
-    action = DummyAction(args=args)
-    context = DummyContext()
-
-    result = GenerateQAPlot(action, context)._perform(args, config={})
-    return result["output_path"]
+    output_path = generate_qa_plot(data, output_path, title)
+    return output_path

src/workflows/prefect_tasks/save_corrected.py

@@ -1,19 +1,8 @@
 from prefect import task
-from keckdrpframework.models.arguments import Arguments
-from keck_primitives.save_corrected import SaveCorrectedFits
-from keck_primitives.utils import DummyAction, DummyContext
+from core.flat import save_corrected_fits
 
 
 @task(name="Save Corrected FITS")
 def save_corrected_fits_task(original_data, correction, header, output_path: str):
-    args = Arguments()
-    args["original_data"] = original_data
-    args["correction"] = correction
-    args["header"] = header
-    args["output_path"] = output_path
-
-    action = DummyAction(args=args)
-    context = DummyContext()
-
-    result = SaveCorrectedFits(action, context)._perform(args, config={})
-    return result["output_path"]
+    output_path = save_corrected_fits(original_data, correction, header, output_path)
+    return output_path

src/workflows/prefect_tasks/save_trace.py

@@ -1,17 +1,8 @@
 from prefect import task
-from keckdrpframework.models.arguments import Arguments
-from keck_primitives.save_trace import SaveTraceSolution
-from keck_primitives.utils import DummyAction, DummyContext
+from core.tracing import save_trace_solution
 
 
 @task(name="Save Trace Solution")
 def save_trace_solution_task(slit_positions, output_path: str):
-    args = Arguments()
-    args["slit_positions"] = slit_positions
-    args["output_path"] = output_path
-
-    action = DummyAction(args=args)
-    context = DummyContext()
-
-    result = SaveTraceSolution(action, context)._perform(args, config={})
-    return result["output_path"]
+    output_path = save_trace_solution(slit_positions, output_path)
+    return output_path

src/workflows/prefect_tasks/trace_slits.py

@@ -1,14 +1,9 @@
 from prefect import task
-from keck_primitives.trace_slits import TraceSlits1D
-from keckdrpframework.models.arguments import Arguments
-from keck_primitives.utils import DummyAction, DummyContext
+from core.tracing import trace_slits_1d
+
 
 @task(name="Trace Slits 1D")
 def trace_slits_task(data):
     """Task to trace slits in 1D data."""
-    args = Arguments()
-    args["flat_data"] = data
-    action = DummyAction(args=args)
-    context = DummyContext()
-    result = TraceSlits1D(action, context)._perform(args, config={})
-    return result["slit_positions"]
+    slit_positions = trace_slits_1d(data)
+    return slit_positions