Skip to content

Commit 4eea26d

Browse files
chcostGitHub Enterprise
authored andcommitted
Merge branch 'develop' into lineage_semantics
2 parents 47b7d1f + 0408629 commit 4eea26d

File tree

5 files changed

+35
-5
lines changed

5 files changed

+35
-5
lines changed

.github/pull_request_template.md

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
**Related Issue**
2+
3+
<!-- List any issues that this pr is related to, fixes or supports -->
4+
5+
Supports #ISSUE_NUMBER
6+
7+
**Related PRs**
8+
9+
<!-- Does this pr depend on or replace an existing pr? Link it here -->
10+
11+
This PR is not dependent on any other PR
12+
13+
**What does this PR do?**
14+
15+
<!-- The intent of this section is to help team members understand the scope of your changes and why you are making said changes. If there are any concerns or risks associated with your pull request, they should be documented here so your team members can give informative feedback in their reviews. -->
16+
17+
**Description of Changes**
18+
19+
<!-- The intent of this section is to describe the implementation details of the changes you made. Any architectural decisions made should be noted here as well as a short description of why you made that decision. -->
20+
21+
**What gif most accurately describes how I feel towards this PR?**
22+
23+
<!-- Cute animals are also acceptable. -->
24+
25+
![Example of a gif](https://media.github.ibm.com/user/59/files/074da280-78ba-11ea-81d1-49ce0654c29d)

codeflare/pipelines/Datamodel.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@
77
from sklearn.base import TransformerMixin
88
from sklearn.base import BaseEstimator
99

10-
1110
class Xy:
1211
"""
1312
Holder class for Xy, where X is array-like and y is array-like. This is the base
@@ -93,6 +92,7 @@ class Node(ABC):
9392
The hash code of this node is the name of the node and equality is defined if the
9493
node name and the type of the node match.
9594
"""
95+
9696
def __init__(self, node_name, node_input_type: NodeInputType, node_firing_type: NodeFiringType, node_state_type: NodeStateType):
9797
self.__node_name__ = node_name
9898
self.__node_input_type__ = node_input_type
@@ -125,6 +125,7 @@ def __hash__(self):
125125
126126
:return: Hash code
127127
"""
128+
128129
return self.__id__.__hash__()
129130

130131
def __eq__(self, other):
@@ -155,6 +156,7 @@ def __init__(self, node_name: str, estimator: BaseEstimator):
155156
:param node_name: Name of the node
156157
:param estimator: The base estimator
157158
"""
159+
158160
super().__init__(node_name, NodeInputType.OR, NodeFiringType.ANY, NodeStateType.IMMUTABLE)
159161
self.__estimator__ = estimator
160162

codeflare/pipelines/Runtime.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import ray
22

3+
34
from codeflare.pipelines.Datamodel import EstimatorNode
45
from codeflare.pipelines.Datamodel import AndNode
56
from codeflare.pipelines.Datamodel import Edge
@@ -64,6 +65,7 @@ def execute_or_node_remote(node: EstimatorNode, train_mode: ExecutionType, xy_re
6465
res_Xref = ray.put(cloned_estimator.fit_transform(X, y))
6566
curr_node_ptr = ray.put(cloned_node)
6667
result = XYRef(res_Xref, xy_ref.get_yref(), prev_node_ptr, curr_node_ptr, [xy_ref])
68+
6769
return result
6870
elif train_mode == ExecutionType.PREDICT:
6971
# Test mode does not clone as it is a simple predict or transform

notebooks/Ray graph experiments.ipynb

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@
137137
},
138138
{
139139
"cell_type": "code",
140-
"execution_count": 9,
140+
"execution_count": 34,
141141
"id": "uniform-coordinator",
142142
"metadata": {},
143143
"outputs": [],
@@ -148,7 +148,7 @@
148148
},
149149
{
150150
"cell_type": "code",
151-
"execution_count": 10,
151+
"execution_count": 35,
152152
"id": "romance-sender",
153153
"metadata": {},
154154
"outputs": [],
@@ -166,7 +166,7 @@
166166
},
167167
{
168168
"cell_type": "code",
169-
"execution_count": 11,
169+
"execution_count": 36,
170170
"id": "handy-offset",
171171
"metadata": {},
172172
"outputs": [],

requirements.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
ray~=1.0.0
22
setuptools~=52.0.0
33
sklearn~=0.0
4-
scikit-learn~=0.24.1
4+
scikit-learn~=0.24.1
5+
pandas~=1.2.4

0 commit comments

Comments
 (0)