made spatialcolortransform compat with type()

1 files changed, 8 insertions, 0 deletions

diff --git a/SpatialColorTransform.lua b/SpatialColorTransform.lua
index 2179b0c..a06efd4 100644
--- a/SpatialColorTransform.lua
+++ b/SpatialColorTransform.lua
@@ -122,6 +122,13 @@ function SpatialColorTransform:backward(input, gradOutput)
    return self.gradInput
 end
 
+function SpatialColorTransform:type(type)
+   parent.type(self,type)
+   if self.islinear then
+      self.linear:type(type)
+   end
+end
+
 function SpatialColorTransform:write(file)
    parent.write(self, file)
    file:writeObject(self.transform)
@@ -138,4 +145,5 @@ function SpatialColorTransform:read(file)
    if self.islinear then
       self.linear = file:readObject()
    end
+   self.type = nil
 end