From 6428557f47cd36e4f0a0b0644fc78e97e5b6431d Mon Sep 17 00:00:00 2001
From: Ryan Curtin <ryan@ratml.org>
Date: Fri, 18 Jun 2021 20:57:25 -0400
Subject: [PATCH] Include ensmallen so ENS_VERSION_MAJOR is defined.

---
 src/mlpack/methods/reinforcement_learning/q_learning.hpp         | 1 +
 src/mlpack/methods/reinforcement_learning/sac.hpp                | 1 +
 .../reinforcement_learning/worker/n_step_q_learning_worker.hpp   | 1 +
 .../reinforcement_learning/worker/one_step_q_learning_worker.hpp | 1 +
 .../reinforcement_learning/worker/one_step_sarsa_worker.hpp      | 1 +
 5 files changed, 5 insertions(+)

diff --git a/src/mlpack/methods/reinforcement_learning/q_learning.hpp b/src/mlpack/methods/reinforcement_learning/q_learning.hpp
index afbdcff33e..3e03bb8c40 100644
--- a/src/mlpack/methods/reinforcement_learning/q_learning.hpp
+++ b/src/mlpack/methods/reinforcement_learning/q_learning.hpp
@@ -14,6 +14,7 @@
 #define MLPACK_METHODS_RL_Q_LEARNING_HPP
 
 #include <mlpack/prereqs.hpp>
+#include <ensmallen.hpp>
 
 #include "replay/random_replay.hpp"
 #include "replay/prioritized_replay.hpp"
diff --git a/src/mlpack/methods/reinforcement_learning/sac.hpp b/src/mlpack/methods/reinforcement_learning/sac.hpp
index f0431185f8..27ddb5cba0 100644
--- a/src/mlpack/methods/reinforcement_learning/sac.hpp
+++ b/src/mlpack/methods/reinforcement_learning/sac.hpp
@@ -14,6 +14,7 @@
 #define MLPACK_METHODS_RL_SAC_HPP
 
 #include <mlpack/prereqs.hpp>
+#include <ensmallen.hpp>
 
 #include "replay/random_replay.hpp"
 #include <mlpack/methods/ann/activation_functions/tanh_function.hpp>
diff --git a/src/mlpack/methods/reinforcement_learning/worker/n_step_q_learning_worker.hpp b/src/mlpack/methods/reinforcement_learning/worker/n_step_q_learning_worker.hpp
index b332752d64..4b052ec892 100644
--- a/src/mlpack/methods/reinforcement_learning/worker/n_step_q_learning_worker.hpp
+++ b/src/mlpack/methods/reinforcement_learning/worker/n_step_q_learning_worker.hpp
@@ -13,6 +13,7 @@
 #ifndef MLPACK_METHODS_RL_WORKER_N_STEP_Q_LEARNING_WORKER_HPP
 #define MLPACK_METHODS_RL_WORKER_N_STEP_Q_LEARNING_WORKER_HPP
 
+#include <ensmallen.hpp>
 #include <mlpack/methods/reinforcement_learning/training_config.hpp>
 
 namespace mlpack {
diff --git a/src/mlpack/methods/reinforcement_learning/worker/one_step_q_learning_worker.hpp b/src/mlpack/methods/reinforcement_learning/worker/one_step_q_learning_worker.hpp
index a85d7cb13a..6916e22185 100644
--- a/src/mlpack/methods/reinforcement_learning/worker/one_step_q_learning_worker.hpp
+++ b/src/mlpack/methods/reinforcement_learning/worker/one_step_q_learning_worker.hpp
@@ -13,6 +13,7 @@
 #ifndef MLPACK_METHODS_RL_WORKER_ONE_STEP_Q_LEARNING_WORKER_HPP
 #define MLPACK_METHODS_RL_WORKER_ONE_STEP_Q_LEARNING_WORKER_HPP
 
+#include <ensmallen.hpp>
 #include <mlpack/methods/reinforcement_learning/training_config.hpp>
 
 namespace mlpack {
diff --git a/src/mlpack/methods/reinforcement_learning/worker/one_step_sarsa_worker.hpp b/src/mlpack/methods/reinforcement_learning/worker/one_step_sarsa_worker.hpp
index 95451d30d4..1b114a1bde 100644
--- a/src/mlpack/methods/reinforcement_learning/worker/one_step_sarsa_worker.hpp
+++ b/src/mlpack/methods/reinforcement_learning/worker/one_step_sarsa_worker.hpp
@@ -13,6 +13,7 @@
 #ifndef MLPACK_METHODS_RL_WORKER_ONE_STEP_SARSA_WORKER_HPP
 #define MLPACK_METHODS_RL_WORKER_ONE_STEP_SARSA_WORKER_HPP
 
+#include <ensmallen.hpp>
 #include <mlpack/methods/reinforcement_learning/training_config.hpp>
 
 namespace mlpack {