add rul support

scripts/createdataset.py

@@ -6,6 +6,7 @@ import tarfile
 from chargefile import ChargeFile
 from spectrafile import SpectraFile
 from soc_estimation import add_soc_estimate
+from rul import add_rul
 
 
 if __name__ == "__main__":
@@ -53,6 +54,7 @@ if __name__ == "__main__":
 	spectras.sort(key=lambda spectra: spectra.step)
 
 	add_soc_estimate(spectras)
+	add_rul(spectras, cells)
 
 	for spectra in spectras:
 		spectra.write(args.out + ".tmp")