@@ -23,7 +23,7 @@ namespace Unity.MLAgentsExamples
23
23
/// </summary>
24
24
public class ModelOverrider : MonoBehaviour
25
25
{
26
- HashSet < string > k_SupportedExtensions = new HashSet < string > { "nn" , "onnx" } ;
26
+ HashSet < string > k_SupportedExtensions = new HashSet < string > { "nn" , "onnx" } ;
27
27
const string k_CommandLineModelOverrideFlag = "--mlagents-override-model" ;
28
28
const string k_CommandLineModelOverrideDirectoryFlag = "--mlagents-override-model-directory" ;
29
29
const string k_CommandLineModelOverrideExtensionFlag = "--mlagents-override-model-extension" ;
@@ -63,7 +63,7 @@ public class ModelOverrider : MonoBehaviour
63
63
64
64
int TotalCompletedEpisodes
65
65
{
66
- get { return m_PreviousAgentCompletedEpisodes + ( m_Agent == null ? 0 : m_Agent . CompletedEpisodes ) ; }
66
+ get { return m_PreviousAgentCompletedEpisodes + ( m_Agent == null ? 0 : m_Agent . CompletedEpisodes ) ; }
67
67
}
68
68
69
69
int TotalNumSteps
@@ -73,7 +73,7 @@ int TotalNumSteps
73
73
74
74
public bool HasOverrides
75
75
{
76
- get { return m_BehaviorNameOverrides . Count > 0 || ! string . IsNullOrEmpty ( m_BehaviorNameOverrideDirectory ) ; }
76
+ get { return m_BehaviorNameOverrides . Count > 0 || ! string . IsNullOrEmpty ( m_BehaviorNameOverrideDirectory ) ; }
77
77
}
78
78
79
79
public static string GetOverrideBehaviorName ( string originalBehaviorName )
@@ -99,17 +99,17 @@ void GetAssetPathFromCommandLine()
99
99
var args = commandLineArgsOverride ?? Environment . GetCommandLineArgs ( ) ;
100
100
for ( var i = 0 ; i < args . Length ; i ++ )
101
101
{
102
- if ( args [ i ] == k_CommandLineModelOverrideFlag && i < args . Length - 2 )
102
+ if ( args [ i ] == k_CommandLineModelOverrideFlag && i < args . Length - 2 )
103
103
{
104
104
var key = args [ i + 1 ] . Trim ( ) ;
105
105
var value = args [ i + 2 ] . Trim ( ) ;
106
106
m_BehaviorNameOverrides [ key ] = value ;
107
107
}
108
- else if ( args [ i ] == k_CommandLineModelOverrideDirectoryFlag && i < args . Length - 1 )
108
+ else if ( args [ i ] == k_CommandLineModelOverrideDirectoryFlag && i < args . Length - 1 )
109
109
{
110
110
m_BehaviorNameOverrideDirectory = args [ i + 1 ] . Trim ( ) ;
111
111
}
112
- else if ( args [ i ] == k_CommandLineModelOverrideExtensionFlag && i < args . Length - 1 )
112
+ else if ( args [ i ] == k_CommandLineModelOverrideExtensionFlag && i < args . Length - 1 )
113
113
{
114
114
m_OverrideExtension = args [ i + 1 ] . Trim ( ) . ToLower ( ) ;
115
115
var isKnownExtension = k_SupportedExtensions . Contains ( m_OverrideExtension ) ;
@@ -124,7 +124,7 @@ void GetAssetPathFromCommandLine()
124
124
#endif
125
125
}
126
126
}
127
- else if ( args [ i ] == k_CommandLineQuitAfterEpisodesFlag && i < args . Length - 1 )
127
+ else if ( args [ i ] == k_CommandLineQuitAfterEpisodesFlag && i < args . Length - 1 )
128
128
{
129
129
Int32 . TryParse ( args [ i + 1 ] , out maxEpisodes ) ;
130
130
}
@@ -198,7 +198,7 @@ public NNModel GetModelForBehaviorName(string behaviorName)
198
198
{
199
199
assetPath = m_BehaviorNameOverrides [ behaviorName ] ;
200
200
}
201
- else if ( ! string . IsNullOrEmpty ( m_BehaviorNameOverrideDirectory ) )
201
+ else if ( ! string . IsNullOrEmpty ( m_BehaviorNameOverrideDirectory ) )
202
202
{
203
203
assetPath = Path . Combine ( m_BehaviorNameOverrideDirectory , $ "{ behaviorName } .{ m_OverrideExtension } ") ;
204
204
}
@@ -214,7 +214,7 @@ public NNModel GetModelForBehaviorName(string behaviorName)
214
214
{
215
215
model = File . ReadAllBytes ( assetPath ) ;
216
216
}
217
- catch ( IOException )
217
+ catch ( IOException )
218
218
{
219
219
Debug . Log ( $ "Couldn't load file { assetPath } at full path { Path . GetFullPath ( assetPath ) } ", this ) ;
220
220
// Cache the null so we don't repeatedly try to load a missing file
@@ -270,7 +270,7 @@ void OverrideModel()
270
270
271
271
if ( ! overrideOk && m_QuitOnLoadFailure )
272
272
{
273
- if ( ! string . IsNullOrEmpty ( overrideError ) )
273
+ if ( ! string . IsNullOrEmpty ( overrideError ) )
274
274
{
275
275
Debug . LogWarning ( overrideError ) ;
276
276
}
0 commit comments