PipelineTask¶
- class lsst.pipe.base.PipelineTask(*, config: PipelineTaskConfig | None = None, log: logging.Logger | LsstLogAdapter | None = None, initInputs: dict[str, Any] | None = None, **kwargs: Any)¶
Bases:
Task
Base class for all pipeline tasks.
This is an abstract base class for PipelineTasks which represents an algorithm executed by framework(s) on data which comes from data butler, resulting data is also stored in a data butler.
PipelineTask inherits from a
pipe.base.Task
and uses the same configuration mechanism based onpex.config
.PipelineTask
classes also have aPipelineTaskConnections
class associated with their config which defines all of the IO aPipelineTask
will need to do. PipelineTask sub-class typically implementsrun()
method which receives Python-domain data objects and returnspipe.base.Struct
object with resulting data.run()
method is not supposed to perform any I/O, it operates entirely on in-memory objects.runQuantum()
is the method (can be re-implemented in sub-class) where all necessary I/O is performed, it reads all input data from data butler into memory, callsrun()
method with that data, examines returnedStruct
object and saves some or all of that data back to data butler.runQuantum()
method receives aQuantumContext
instance to facilitate I/O, aInputQuantizedConnection
instance which defines all inputlsst.daf.butler.DatasetRef
, and aOutputQuantizedConnection
instance which defines all the outputlsst.daf.butler.DatasetRef
for a single invocation of PipelineTask.Subclasses must be constructable with exactly the arguments taken by the PipelineTask base class constructor, but may support other signatures as well.
- Parameters:
- config
pex.config.Config
, optional Configuration for this task (an instance of
self.ConfigClass
, which is a task-specific subclass ofPipelineTaskConfig
). If not specified then it defaults toself.ConfigClass()
.- log
logging.Logger
, optional Logger instance whose name is used as a log name prefix, or
None
for no prefix.- initInputs
dict
, optional A dictionary of objects needed to construct this PipelineTask, with keys matching the keys of the dictionary returned by
getInitInputDatasetTypes
and values equivalent to what would be obtained by callingButler.get
with those DatasetTypes and no data IDs. While it is optional for the base class, subclasses are permitted to require this argument.
- config
- Attributes:
- canMultiprocessbool, True by default (class attribute)
This class attribute is checked by execution framework, sub-classes can set it to
False
in case task does not support multiprocessing.
Attributes Summary
Methods Summary
Empty (clear) the metadata for this Task and all sub-Tasks.
Get metadata for all tasks.
Get the task name as a hierarchical name including parent task names.
getName
()Get the name of the task.
Get a dictionary of all tasks as a shallow copy.
makeField
(doc)Make a
lsst.pex.config.ConfigurableField
for this task.makeSubtask
(name, **keyArgs)Create a subtask as a new instance as the
name
attribute of this task.run
(**kwargs)Run task algorithm on in-memory data.
runQuantum
(butlerQC, inputRefs, outputRefs)Do butler IO and transform to provide in memory objects for tasks
run
method.timer
(name[, logLevel])Context manager to log performance data for an arbitrary block of code.
Attributes Documentation
Methods Documentation
- getFullMetadata() TaskMetadata ¶
Get metadata for all tasks.
- Returns:
- metadata
TaskMetadata
The keys are the full task name. Values are metadata for the top-level task and all subtasks, sub-subtasks, etc.
- metadata
Notes
The returned metadata includes timing information (if
@timer.timeMethod
is used) and any metadata set by the task. The name of each item consists of the full task name with.
replaced by:
, followed by.
and the name of the item, e.g.:topLevelTaskName:subtaskName:subsubtaskName.itemName
using
:
in the full task name disambiguates the rare situation that a task has a subtask and a metadata item with the same name.
- getFullName() str ¶
Get the task name as a hierarchical name including parent task names.
- Returns:
- fullName
str
The full name consists of the name of the parent task and each subtask separated by periods. For example:
The full name of top-level task “top” is simply “top”.
The full name of subtask “sub” of top-level task “top” is “top.sub”.
The full name of subtask “sub2” of subtask “sub” of top-level task “top” is “top.sub.sub2”.
- fullName
- getTaskDict() dict[str, weakref.ReferenceType[lsst.pipe.base.task.Task]] ¶
Get a dictionary of all tasks as a shallow copy.
- Returns:
- taskDict
dict
Dictionary containing full task name: task object for the top-level task and all subtasks, sub-subtasks, etc.
- taskDict
- classmethod makeField(doc: str) ConfigurableField ¶
Make a
lsst.pex.config.ConfigurableField
for this task.- Parameters:
- doc
str
Help text for the field.
- doc
- Returns:
- configurableField
lsst.pex.config.ConfigurableField
A
ConfigurableField
for this task.
- configurableField
Examples
Provides a convenient way to specify this task is a subtask of another task.
Here is an example of use:
class OtherTaskConfig(lsst.pex.config.Config): aSubtask = ATaskClass.makeField("brief description of task")
- makeSubtask(name: str, **keyArgs: Any) None ¶
Create a subtask as a new instance as the
name
attribute of this task.- Parameters:
- name
str
Brief name of the subtask.
- **keyArgs
Extra keyword arguments used to construct the task. The following arguments are automatically provided and cannot be overridden:
config
.parentTask
.
- name
Notes
The subtask must be defined by
Task.config.name
, an instance ofConfigurableField
orRegistryField
.
- run(**kwargs: Any) Struct ¶
Run task algorithm on in-memory data.
This method should be implemented in a subclass. This method will receive keyword arguments whose names will be the same as names of connection fields describing input dataset types. Argument values will be data objects retrieved from data butler. If a dataset type is configured with
multiple
field set toTrue
then the argument value will be a list of objects, otherwise it will be a single object.If the task needs to know its input or output DataIds then it has to override
runQuantum
method instead.This method should return a
Struct
whose attributes share the same name as the connection fields describing output dataset types.- Returns:
- struct
Struct
Struct with attribute names corresponding to output connection fields
- struct
Examples
Typical implementation of this method may look like:
def run(self, input, calib): # "input", "calib", and "output" are the names of the config # fields # Assuming that input/calib datasets are `scalar` they are # simple objects, do something with inputs and calibs, produce # output image. image = self.makeImage(input, calib) # If output dataset is `scalar` then return object, not list return Struct(output=image)
- runQuantum(butlerQC: QuantumContext, inputRefs: InputQuantizedConnection, outputRefs: OutputQuantizedConnection) None ¶
Do butler IO and transform to provide in memory objects for tasks
run
method.- Parameters:
- butlerQC
QuantumContext
A butler which is specialized to operate in the context of a
lsst.daf.butler.Quantum
.- inputRefs
InputQuantizedConnection
Datastructure whose attribute names are the names that identify connections defined in corresponding
PipelineTaskConnections
class. The values of these attributes are thelsst.daf.butler.DatasetRef
objects associated with the defined input/prerequisite connections.- outputRefs
OutputQuantizedConnection
Datastructure whose attribute names are the names that identify connections defined in corresponding
PipelineTaskConnections
class. The values of these attributes are thelsst.daf.butler.DatasetRef
objects associated with the defined output connections.
- butlerQC