refactor call for Gyx.Gym.Utils.gyx_space/1 #26¨

doctorcorral · Mar 1, 2019 · 2b3d62a · 2b3d62a
1 parent 1a74f20
commit 2b3d62a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/lib/Gym/environment.ex b/lib/Gym/environment.ex
@@ -5,7 +5,7 @@ defmodule Gyx.Gym.Environment do
   """
   alias Gyx.Python.HelperAsync
   alias Gyx.Core.{Env, Exp}
-  alias Gyx.Gym.Utils
+  import Gyx.Gym.Utils, only: [gyx_space: 1]
   use Env
   use GenServer
   require Logger
@@ -65,7 +65,7 @@ defmodule Gyx.Gym.Environment do
        env: env,
        current_state: initial_state,
        session: state.session,
-       action_space: Utils.gyx_space(action_space)
+       action_space: gyx_space(action_space)
      }}
   end
 

diff --git a/lib/Gym/space_handler.ex b/lib/Gym/space_handler.ex
@@ -14,7 +14,7 @@ defmodule Gyx.Gym.Utils do
     gym_space_string |> parse |> create_space
   end
 
-  def parse(gym_space_string) do
+  defp parse(gym_space_string) do
     {regex, type} =
       Enum.find(@space_types, {nil, :unknown}, fn {regex, _} ->
         Regex.match?(regex, Kernel.inspect(gym_space_string))