From 8bdd5e2132e08c0166b2e77c7de807fde0fbfcde Mon Sep 17 00:00:00 2001
From: Pulkit gera <pulkithifi@yahoo.com>
Date: Sun, 27 Oct 2019 04:02:57 +0530
Subject: [PATCH] added docstring
---
flatland/envs/rail_generators.py | 11 +++++++++++
1 file changed, 11 insertions(+)
diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py
index 028928ae..3e90128c 100644
--- a/flatland/envs/rail_generators.py
+++ b/flatland/envs/rail_generators.py
@@ -343,6 +343,17 @@ def random_rail_generator(cell_type_relative_proportion=[1.0] * 11, seed=1) -> R
template = [template[-1]] + template[:-1]
def get_matching_templates(template):
+ """
+ Returns a list of possible transition maps for a given template
+
+ Parameters:
+ ------
+ template:List[int]
+
+ Returns:
+ ------
+ List[int]
+ """
ret = []
for i in range(len(transitions_templates_)):
is_match = True
--
GitLab