start writing tests

vpuri3 · May 12, 2024 · 55fec4a · 55fec4a
1 parent 4bd7aa0
commit 55fec4a
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/Project.toml b/Project.toml
@@ -16,6 +16,10 @@ julia = "1.6"
 
 [extras]
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
+Lux = "b2108857-7c20-44ae-9111-449ecde12c47"
+Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
+Optimisers = "3bd65402-5787-11e9-1adc-39752487f4e2"
+OptimizationOptimJL = "36348300-93cb-4f02-beb5-3c3902f8871e"
 
 [targets]
-test = ["Test"]
+test = ["Test", "Lux", "Zygote", "Optimisers", "OptimizationOptimJL"]
diff --git a/src/type.jl b/src/type.jl
@@ -1,4 +1,9 @@
 #
+#======================================================#
+# Kolmogorov-Arnold Layer
+# https://github.com/mintisan/awesome-kan
+# https://github.com/ZiyaoLi/fast-kan/tree/master
+#======================================================#
 @concrete struct KDense <: LuxCore.AbstractExplicitLayer
     in_dims::Int
     out_dims::Int

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -1,6 +1,9 @@
 using KolmogorovArnold
 using Test
 
+using Lux, Zygote
+using Optimisers, OptimizationOptimJL
+
 @testset "KolmogorovArnold.jl" begin
     # Write your tests here.
 end