From 36d36ed47102d5e31f1855aaf17403d8a956a856 Mon Sep 17 00:00:00 2001
From: Junhao Liang <43094337+darkliang@users.noreply.github.com>
Date: Wed, 27 Sep 2023 17:35:20 +0800
Subject: [PATCH] fix var name

---
 opengait/data/transform.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/opengait/data/transform.py b/opengait/data/transform.py
index c5fd0d4..065c7aa 100644
--- a/opengait/data/transform.py
+++ b/opengait/data/transform.py
@@ -236,10 +236,10 @@ class MirrorPoses(object):
     Performing Mirror Operations
     """
     def __init__(self, prob=0.5):
-        self.probability = probability
+        self.prob = prob
 
     def __call__(self, data):
-        if np.random.random() <= self.probability:
+        if np.random.random() <= self.prob:
             center = np.mean(data[:, :, 0], axis=1, keepdims=True)
             data[:, :, 0] = center - data[:, :, 0] + center
 
@@ -449,4 +449,4 @@ class MSGGTransform():
         
     def __call__(self, x):
         result=x[...,self.mask,:].copy()
-        return result
\ No newline at end of file
+        return result