From 84a3bd2826f975d11412664e6730f52b0cdc5d84 Mon Sep 17 00:00:00 2001
From: moayadeldin <moayadeldin>
Date: Sat, 15 Jun 2024 00:44:11 +0300
Subject: [PATCH] Fix tensorflow-gpu and Keras Preprocessing dependencies issue

---
 aranet/aranet.py | 2 +-
 setup.py         | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/aranet/aranet.py b/aranet/aranet.py
index 41c6487..3e4a3ed 100644
--- a/aranet/aranet.py
+++ b/aranet/aranet.py
@@ -3,7 +3,7 @@
 import os, json
 import numpy as np
 import pandas as pd
-from keras.preprocessing.sequence import pad_sequences
+from keras_preprocessing.sequence import pad_sequences
 import torch
 from torch import nn
 from torch.utils.data import DataLoader, TensorDataset
diff --git a/setup.py b/setup.py
index 47d99cf..ba35851 100644
--- a/setup.py
+++ b/setup.py
@@ -9,10 +9,10 @@
       license='GNU',
       packages=['aranet'],
       install_requires=[
-          'tensorflow-gpu',
+          'tensorflow',
           'torch',
           'sklearn',
-          'transformers==2.3.0', 'keras', 'pandas', 'numpy'
+          'transformers==2.3.0', 'keras', 'pandas', 'numpy', 'Keras-Preprocessing'
       ],
 
       entry_points={