rename GraphMethod type aliases

Author: EronWright

flink-tensorflow/src/main/scala/org/apache/flink/contrib/tensorflow/examples/inception/ImageNormalization.scala

@@ -13,8 +13,8 @@ import org.tensorflow.framework.{SignatureDef, TensorInfo}
 
 sealed trait ImageNormalizationMethod extends GraphMethod {
   val name = NORMALIZE_METHOD_NAME
-  override type IN = ImageFileTensor
-  override type OUT = ImageTensor
+  override type Input = ImageFileTensor
+  override type Output = ImageTensor
 }
 
 object ImageNormalizationMethod {
@@ -26,8 +26,8 @@ object ImageNormalizationMethod {
     * Normalizes a vector of image files to a vector of images.
     */
   implicit val impl = new ImageNormalizationMethod {
-    def inputs(i: IN): Map[String, Tensor] = Map(NORMALIZE_INPUTS -> i)
-    def outputs(o: Map[String, Tensor]): OUT = o(NORMALIZE_OUTPUTS).taggedAs[ImageTensor]
+    def inputs(i: Input): Map[String, Tensor] = Map(NORMALIZE_INPUTS -> i)
+    def outputs(o: Map[String, Tensor]): Output = o(NORMALIZE_OUTPUTS).taggedAs[ImageTensor]
   }
 }
 

flink-tensorflow/src/main/scala/org/apache/flink/contrib/tensorflow/examples/inception/InceptionModel.scala

@@ -18,8 +18,8 @@ import scala.collection.JavaConverters._
 
 sealed trait LabelMethod extends GraphMethod {
   def name = LABEL_METHOD_NAME
-  override type IN = ImageTensor
-  override type OUT = LabelTensor
+  override type Input = ImageTensor
+  override type Output = LabelTensor
 }
 
 @SerialVersionUID(1L)

flink-tensorflow/src/main/scala/org/apache/flink/contrib/tensorflow/graphs/GraphMethod.scala

@@ -10,12 +10,12 @@ trait GraphMethod {
   /**
     * The input type of the method.
     */
-  type IN
+  type Input
 
   /**
     * The output type of the method.
     */
-  type OUT
+  type Output
 
   /**
     * The method name.
@@ -25,12 +25,12 @@ trait GraphMethod {
   /**
     * Gets the input values to feed when the method is invoked.
     */
-  def inputs(in: IN): Map[String, Tensor]
+  def inputs(in: Input): Map[String, Tensor]
 
   /**
     * Gets the result of invoking the method.
     * @param outputs a map of fetched outputs.
     */
-  def outputs(outputs: Map[String, Tensor]): OUT
+  def outputs(outputs: Map[String, Tensor]): Output
 
 }

flink-tensorflow/src/main/scala/org/apache/flink/contrib/tensorflow/ml/signatures/RegressionMethod.scala

@@ -17,8 +17,8 @@ import org.apache.flink.contrib.tensorflow.graphs.GraphMethod
   */
 sealed trait RegressionMethod extends GraphMethod {
   val name = REGRESS_METHOD_NAME
-  override type IN = ExampleTensor
-  override type OUT = PredictionTensor
+  override type Input = ExampleTensor
+  override type Output = PredictionTensor
 }
 
 object RegressionMethod {

flink-tensorflow/src/main/scala/org/apache/flink/contrib/tensorflow/models/ModelFunction.scala

@@ -21,7 +21,7 @@ trait ModelFunction[T <: GraphMethod] {
     *
     * @return the output values as provided by the method.
     */
-  def apply(in: T#IN): ManagedResource[T#OUT]
+  def apply(in: T#Input): ManagedResource[T#Output]
 }
 
 object ModelFunction {
@@ -37,7 +37,7 @@ object ModelFunction {
     require(method.name == signatureDef.getMethodName)
 
     new ModelFunction[T] {
-      def apply(in: T#IN): ManagedResource[T#OUT] = {
+      def apply(in: T#Input): ManagedResource[T#Output] = {
         require(in != null, "Input must be provided")
 
         // create a managed resource that lazily runs the graph
@@ -47,7 +47,7 @@ object ModelFunction {
             val runner = session.runner
 
             // map inputs according to the signaturedef
-            val inputs = method.inputs(in.asInstanceOf[method.IN])
+            val inputs = method.inputs(in.asInstanceOf[method.Input])
             signatureDef.getInputsMap.asScala.foreach { kv =>
               val tensor = inputs.getOrElse(kv._1,
                 throw new IllegalArgumentException(s"An input tensor named ${kv._1} must be provided for this computation."))