From 65cddab6149b089e6e4b8b60c0dbc6759c41a61f Mon Sep 17 00:00:00 2001 From: Samuel William Nehrer Date: Mon, 29 Jan 2024 10:27:12 +0100 Subject: [PATCH] Env-- --- src/ActiveInference.jl | 10 ++++------ src/environment.jl | 1 - 2 files changed, 4 insertions(+), 7 deletions(-) diff --git a/src/ActiveInference.jl b/src/ActiveInference.jl index 188b136..d7d28aa 100644 --- 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 index 4af858d..70ed20e 100644 --- a/src/environment.jl +++ b/src/environment.jl @@ -1,5 +1,4 @@ using LinearAlgebra -using Plots using IterTools