@NotNull
  Map<String, MLFeatureValue> calculateFeatures(@NotNull Map<String, MLFeatureValue> contextFeatures);
@NotNull
  Map<String, MLFeatureValue> calculateFeatures(@NotNull Map<String, MLFeatureValue> contextFeatures)