Env--

ilabcode · Jan 29, 2024 · 65cddab · 65cddab
1 parent 18e92a7
commit 65cddab
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/src/ActiveInference.jl b/src/ActiveInference.jl
@@ -5,7 +5,7 @@ include("maths.jl")
 include("environment.jl")
 
 # From functions.jl
-export array_of_any, array_of_any_zeros, array_of_any_uniform, plot_beliefs, plot_gridworld, plot_likelihood, create_B_matrix, onehot, plot_point_on_grid, infer_states, get_expected_states, get_expected_observations, calculate_G, run_active_inference_loop, construct_policies, calculate_G_policies, compute_prob_actions,active_inference_with_planning, EpistChainEnv, GridWorldEnv
+export array_of_any, array_of_any_zeros, array_of_any_uniform, plot_beliefs, plot_gridworld, plot_likelihood, create_B_matrix, onehot, plot_point_on_grid, infer_states, get_expected_states, get_expected_observations, calculate_G, run_active_inference_loop, construct_policies, calculate_G_policies, compute_prob_actions,active_inference_with_planning, GridWorldEnv
 
     # From maths.jl
     module Maths
@@ -14,18 +14,16 @@ export array_of_any, array_of_any_zeros, array_of_any_uniform, plot_beliefs, plo
 
     export norm_dist, sample_category, softmax, spm_log_single, entropy, kl_divergence
 
-    end;
+    end
 
     # From environment.jl
     module Environment
 
     include("environment.jl")
 
-    export GridWorldEnv, EpistChainEnv, step!, reset!
+    export EpistChainEnv, step!, reset!
 
-    end;
-
-
+    end
 end
 
 

diff --git a/src/environment.jl b/src/environment.jl
@@ -1,5 +1,4 @@
 using LinearAlgebra
-using Plots
 using IterTools