Fixed state assignment in WaveletKernel

Author: mandar2812

src/main/scala/io/github/mandar2812/dynaml/kernels/WaveletKernel.scala

@@ -1,3 +1,21 @@
+/*
+Licensed to the Apache Software Foundation (ASF) under one
+or more contributor license agreements.  See the NOTICE file
+distributed with this work for additional information
+regarding copyright ownership.  The ASF licenses this file
+to you under the Apache License, Version 2.0 (the
+"License"); you may not use this file except in compliance
+with the License.  You may obtain a copy of the License at
+
+  http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing,
+software distributed under the License is distributed on an
+"AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+KIND, either express or implied.  See the License for the
+specific language governing permissions and limitations
+under the License.
+* */
 package io.github.mandar2812.dynaml.kernels
 
 import breeze.linalg.{DenseVector, DenseMatrix}
@@ -12,6 +30,8 @@ class WaveletKernel(func: (Double) => Double)(private var scale: Double)
 
   override val hyper_parameters = List("scale")
 
+  state = Map("scale" -> scale)
+
   val mother: (Double) => Double = func
 
   def setscale(d: Double): Unit = {