From 12d09eb45f0f32d2252c8d21d94f9574f3b7bfa4 Mon Sep 17 00:00:00 2001
From: Michael Abbott <32575566+mcabbott@users.noreply.github.com>
Date: Tue, 8 Mar 2022 14:05:41 -0500
Subject: [PATCH 1/2] Flux no longer exports params

---
 test/fourierlayer.jl | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/test/fourierlayer.jl b/test/fourierlayer.jl
index 89bfa30..a26af98 100644
--- a/test/fourierlayer.jl
+++ b/test/fourierlayer.jl
@@ -13,13 +13,13 @@ using Test, Random, Flux
     # We only use a subset of the weight tensors for training
     @testset "parameters" begin
         # Wf
-        @test size(params(FourierLayer(128, 64, 100, 20))[1]) == (128, 64, 20)
+        @test size(Flux.params(FourierLayer(128, 64, 100, 20))[1]) == (128, 64, 20)
         # Wl
-        @test size(params(FourierLayer(128, 64, 100, 20))[2]) == (64, 128)
+        @test size(Flux.params(FourierLayer(128, 64, 100, 20))[2]) == (64, 128)
         # bf
-        @test size(params(FourierLayer(128, 64, 100, 20))[3]) == (1, 64, 20)
+        @test size(Flux.params(FourierLayer(128, 64, 100, 20))[3]) == (1, 64, 20)
         # bl
-        @test size(params(FourierLayer(128, 64, 100, 20))[4]) == (1, 64, 100)
+        @test size(Flux.params(FourierLayer(128, 64, 100, 20))[4]) == (1, 64, 100)
     end
 
     # Accept only Int as architecture parameters
@@ -28,4 +28,4 @@ using Test, Random, Flux
     # Test max amount of modes
     @test_throws AssertionError FourierLayer(100, 100, 100, 60, σ)
     @test_throws AssertionError FourierLayer(100, 100, 100, 60)
-end
\ No newline at end of file
+end

From 4727facad47b3079054a5f0189e835a0bb48d9f3 Mon Sep 17 00:00:00 2001
From: Michael Abbott <32575566+mcabbott@users.noreply.github.com>
Date: Tue, 8 Mar 2022 14:07:00 -0500
Subject: [PATCH 2/2] versions

---
 Project.toml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Project.toml b/Project.toml
index 8435bc7..8e9a2d0 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "OperatorLearning"
 uuid = "5913d0e6-5bb6-45e3-8a06-341cf4fd0203"
 authors = ["Patrick Zimbrod <76100656+pzimbrod@users.noreply.github.com>"]
-version = "0.2.1"
+version = "0.2.2"
 
 [deps]
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
@@ -14,7 +14,7 @@ Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 [compat]
 CUDA = "3"
 FFTW = "1"
-Flux = "0.12"
+Flux = "0.12, 0.13"
 NNlib = "0.8"
 OMEinsum = "0.6"
 julia = "1.6"