Pyre Configurationless migration for] [batch:90/244]

Reviewed By: connernilsen

Differential Revision: D54472020

fbshipit-source-id: 0c4ba47ab15a5ef47d5a32aa9afcd90b3cf69491

pearl/__init__.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 from .pearl_agent import PearlAgent
 
 __all__ = ["PearlAgent"]

pearl/action_representation_modules/__init__.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 from .action_representation_module import ActionRepresentationModule
 from .binary_action_representation_module import BinaryActionTensorRepresentationModule
 from .identity_action_representation_module import IdentityActionRepresentationModule

pearl/action_representation_modules/action_representation_module.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 from abc import ABC, abstractmethod
 
 import torch

pearl/action_representation_modules/binary_action_representation_module.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 import torch
 
 from pearl.action_representation_modules.action_representation_module import (

pearl/action_representation_modules/identity_action_representation_module.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 import torch
 from pearl.action_representation_modules.action_representation_module import (
     ActionRepresentationModule,

pearl/action_representation_modules/one_hot_action_representation_module.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 import torch
 import torch.nn.functional as F
 

pearl/api/__init__.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 from .action import Action
 from .action_result import ActionResult
 from .action_space import ActionSpace

pearl/api/action.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 from torch import Tensor
 
 # An `Action` is expected to be a 1-dim Tensor of shape `(d,)`, where `d` is the

pearl/api/action_result.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 from dataclasses import dataclass
 from typing import Any, Dict, Optional
 

pearl/api/action_space.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 from __future__ import annotations
 
 from abc import abstractmethod

pearl/api/agent.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 from abc import ABC, abstractmethod
 
 from pearl.api.action import Action

pearl/api/environment.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 from __future__ import annotations
 
 from abc import ABC, abstractmethod

pearl/api/history.py

@@ -5,4 +5,6 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 History = object

pearl/api/observation.py

@@ -5,4 +5,6 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 Observation = object

pearl/api/reward.py

@@ -5,5 +5,7 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 Value = object
 Reward = Value

pearl/api/space.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 from __future__ import annotations
 
 from abc import ABC, abstractmethod

pearl/api/state.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 from torch import Tensor
 
 SubjectiveState = Tensor

pearl/history_summarization_modules/__init__.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 from .history_summarization_module import HistorySummarizationModule
 from .identity_history_summarization_module import IdentityHistorySummarizationModule
 from .lstm_history_summarization_module import LSTMHistorySummarizationModule

pearl/history_summarization_modules/history_summarization_module.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 from abc import ABC, abstractmethod
 from typing import Optional
 

pearl/history_summarization_modules/identity_history_summarization_module.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 from typing import Optional
 
 import torch

pearl/history_summarization_modules/lstm_history_summarization_module.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 from typing import Optional
 
 import torch

pearl/history_summarization_modules/stacking_history_summarization_module.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 from typing import Optional
 
 import torch

pearl/neural_networks/__init__.py

@@ -3,3 +3,5 @@
 #
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
+
+# pyre-strict

pearl/neural_networks/common/__init__.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 from .epistemic_neural_networks import Ensemble, EpistemicNeuralNetwork, MLPWithPrior
 from .residual_wrapper import ResidualWrapper
 from .value_networks import ValueNetwork, VanillaCNN, VanillaValueNetwork

pearl/neural_networks/common/epistemic_neural_networks.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 """
 This module defines epistemic neural networks that can model posterior distributions
     and perform bayesian updates via deep learning

pearl/neural_networks/common/residual_wrapper.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 import torch
 from torch import nn
 

pearl/neural_networks/common/utils.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 import logging
 from typing import Any, Dict, List, Optional, Union
 

pearl/neural_networks/common/value_networks.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 """
 This module defines several types of value neural networks.
 """

pearl/neural_networks/contextual_bandit/__init__.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 from .base_cb_model import MuSigmaCBModel
 from .linear_regression import LinearRegression
 from .neural_linear_regression import NeuralLinearRegression

pearl/neural_networks/contextual_bandit/base_cb_model.py

@@ -1,3 +1,4 @@
+# pyre-strict
 from abc import abstractmethod
 
 import torch

pearl/neural_networks/contextual_bandit/linear_regression.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 import logging
 from typing import Optional, Tuple
 

pearl/neural_networks/contextual_bandit/neural_linear_regression.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 import logging
 from typing import Dict, List, Optional
 

pearl/neural_networks/sequential_decision_making/__init__.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 from .actor_networks import (
     ActorNetwork,
     DynamicActionActorNetwork,

pearl/neural_networks/sequential_decision_making/actor_networks.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 """
 This module defines several types of actor neural networks.
 """

pearl/neural_networks/sequential_decision_making/q_value_networks.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 """
 This file defines PEARL neural network interafaces
 User is free to define their own Q(s, a), but would need to inherit from this interface

pearl/neural_networks/sequential_decision_making/twin_critic.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 import inspect
 from typing import Callable, Iterable, Tuple, Type
 

pearl/pearl_agent.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 import typing
 from typing import Any, Dict, Optional
 

pearl/policy_learners/__init__.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 from .policy_learner import PolicyLearner
 
 __all__ = ["PolicyLearner"]

pearl/policy_learners/contextual_bandits/__init__.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
 
+# pyre-strict
+
 from .contextual_bandit_base import ContextualBanditBase
 from .disjoint_bandit import DisjointBanditContainer
 from .disjoint_linear_bandit import DisjointLinearBandit

pearl/policy_learners/contextual_bandits/contextual_bandit_base.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 from abc import abstractmethod
 from typing import Any, Dict, Optional
 

pearl/policy_learners/contextual_bandits/disjoint_bandit.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 from typing import Any, Dict, List, Optional
 
 import torch

pearl/policy_learners/contextual_bandits/disjoint_linear_bandit.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 from typing import Any, Dict, List
 
 import torch

pearl/policy_learners/contextual_bandits/linear_bandit.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 from typing import Any, Dict, Optional
 
 import torch

pearl/policy_learners/contextual_bandits/neural_bandit.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 from typing import Any, Dict, List, Optional
 
 import torch

pearl/policy_learners/contextual_bandits/neural_linear_bandit.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 #
 
+# pyre-strict
+
 from typing import Any, Dict, List, Optional
 
 import torch