Upload operator.py with huggingface_hub
Browse files- operator.py +3 -3
operator.py
CHANGED
|
@@ -509,7 +509,7 @@ class SourceSequentialOperator(SequentialOperator):
|
|
| 509 |
return multi_stream
|
| 510 |
|
| 511 |
|
| 512 |
-
class
|
| 513 |
"""A class representing a sequential operator initializer in the streaming system.
|
| 514 |
|
| 515 |
A sequential operator initializer is a type of `SequntialOperator` that starts with a stream initializer operator. The first operator in its list of steps is a `StreamInitializerOperator`, which generates the initial `MultiStream` based on the provided arguments and keyword arguments.
|
|
@@ -521,11 +521,11 @@ class SequentialOperatorInitilizer(SequentialOperator):
|
|
| 521 |
def process(self, *args, **kwargs) -> MultiStream:
|
| 522 |
assert (
|
| 523 |
self.num_steps() > 0
|
| 524 |
-
), "Calling process on a
|
| 525 |
|
| 526 |
assert isinstance(
|
| 527 |
self.steps[0], StreamInitializerOperator
|
| 528 |
-
), "The first step in a
|
| 529 |
multi_stream = self.steps[0](*args, **kwargs)
|
| 530 |
for operator in self.steps[1 : self._get_max_steps()]:
|
| 531 |
multi_stream = operator(multi_stream)
|
|
|
|
| 509 |
return multi_stream
|
| 510 |
|
| 511 |
|
| 512 |
+
class SequentialOperatorInitializer(SequentialOperator):
|
| 513 |
"""A class representing a sequential operator initializer in the streaming system.
|
| 514 |
|
| 515 |
A sequential operator initializer is a type of `SequntialOperator` that starts with a stream initializer operator. The first operator in its list of steps is a `StreamInitializerOperator`, which generates the initial `MultiStream` based on the provided arguments and keyword arguments.
|
|
|
|
| 521 |
def process(self, *args, **kwargs) -> MultiStream:
|
| 522 |
assert (
|
| 523 |
self.num_steps() > 0
|
| 524 |
+
), "Calling process on a SequentialOperatorInitializer without any steps"
|
| 525 |
|
| 526 |
assert isinstance(
|
| 527 |
self.steps[0], StreamInitializerOperator
|
| 528 |
+
), "The first step in a SequentialOperatorInitializer must be a StreamInitializerOperator"
|
| 529 |
multi_stream = self.steps[0](*args, **kwargs)
|
| 530 |
for operator in self.steps[1 : self._get_max_steps()]:
|
| 531 |
multi_stream = operator(multi_stream)
|