1 files changed, 7 insertions, 3 deletions
diff --git a/vest/ces.h b/vest/ces.h
index 2f098990..e021e715 100644
--- a/vest/ces.h
+++ b/vest/ces.h
@@ -1,12 +1,16 @@
 #ifndef _CES_H_
 #define _CES_H_
 
-#include "scorer.h"
-
 class ViterbiEnvelope;
 class Hypergraph;
+class SegmentEvaluator;
 class ErrorSurface;
+class EvaluationMetric;
 
-void ComputeErrorSurface(const SentenceScorer& ss, const ViterbiEnvelope& ve, ErrorSurface* es, const ScoreType type, const Hypergraph& hg);
+void ComputeErrorSurface(const SegmentEvaluator& ss,
+                         const ViterbiEnvelope& ve,
+                         ErrorSurface* es,
+                         const EvaluationMetric* metric,
+                         const Hypergraph& hg);
 
 #endif