Rachel Bawden
commited on
Commit
·
276d772
1
Parent(s):
4ac4212
bug pipeline
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -11,7 +11,7 @@ import operator
|
|
| 11 |
|
| 12 |
|
| 13 |
######## PredTitrage pipeline #########
|
| 14 |
-
class
|
| 15 |
|
| 16 |
def __init__(self, beam_size=5, batch_size=32, **kwargs):
|
| 17 |
self.beam_size = beam_size
|
|
|
|
| 11 |
|
| 12 |
|
| 13 |
######## PredTitrage pipeline #########
|
| 14 |
+
class PredTitragesPipeline(Pipeline):
|
| 15 |
|
| 16 |
def __init__(self, beam_size=5, batch_size=32, **kwargs):
|
| 17 |
self.beam_size = beam_size
|