Skip to content

Commit 7313f78

Browse files
committed
added get_params to StrategySerializer
1 parent 9a31f8e commit 7313f78

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

api/core/serializers.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ class StrategySerializer(serializers.Serializer):
1212
name = serializers.CharField(max_length=200)
1313
description = serializers.SerializerMethodField()
1414
classifier = serializers.SerializerMethodField()
15+
params = serializers.SerializerMethodField()
1516

1617
def get_id(self, strategy):
1718
return strategy_id(strategy)
@@ -31,3 +32,9 @@ def get_classifier(self, strategy):
3132
if classifier['memory_depth'] == float('inf'):
3233
classifier['memory_depth'] = -1
3334
return classifier
35+
36+
def get_params(self, strategy):
37+
params = strategy.init_params()
38+
if 'memory_depth' in params and params['memory_depth'] == float('inf'):
39+
params['memory_depth'] = -1
40+
return params

0 commit comments

Comments
 (0)